Update atomic_facts.py

neulab · Jan 1, 2025 · 62fe55b · 62fe55b
1 parent b7eac00
commit 62fe55b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llments/eval/factscore/atomic_facts.py b/llments/eval/factscore/atomic_facts.py
@@ -8,7 +8,7 @@
 from rank_bm25 import BM25Okapi
 import os
 from nltk.tokenize import sent_tokenize
-from typing import List, Tuple, Optional, Any, Set, Sequence, cast
+from typing import List, Tuple, Optional, Any, Set, cast
 
 from factscore.openai_lm import OpenAIModel
 
@@ -365,7 +365,7 @@ def postprocess_atomic_facts(
     verbs = ["born.", " appointed.", " characterized.", " described.", " known.", " member.", " advocate.", "served.", "elected."]
     permitted_verbs = ["founding member."]
 
-    atomic_facts: List[List[Sequence[str]]] = []
+    atomic_facts: List[Tuple[str, List[str]]] = []
     new_atomic_facts: List[Tuple[str, List[str]]] = []
     new_para_breaks: List[int] = []