Update npm.py

neulab · Jan 1, 2025 · 582a524 · 582a524
1 parent 18e454b
commit 582a524
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llments/eval/factscore/npm.py b/llments/eval/factscore/npm.py
@@ -196,7 +196,7 @@ def get_probabilty(self, topic: str, question: str) -> float:
         passages = [p["text"].strip() for p in passages_2]
         cache_key = question + "#" + "#".join(passages)
         if cache_key not in self.cache_dict:
-            encoded = self.encode(passages, skip_special_tokens=True)
+            encoded = cast(List[Tuple[List[int], np.ndarray]], self.encode(passages, skip_special_tokens=True))
             stacked_passage_tokens: List[int] = []
             stacked_passage_vectors: np.ndarray = []
             for input_ids, vectors in encoded: