claim format debug

minicheck/minicheck.py

@@ -4,7 +4,7 @@ import sys
 sys.path.append("..")
 
 from minicheck.inference import Inferencer
-from typing import List
+from typing import List, Dict
 import numpy as np
 
 
@@ -18,7 +18,7 @@ class MiniCheck:
             max_input_length=max_input_length,
         )
 
-    def score(self, docs: List[str], claims: List[str]) -> List[float]:
+    def score(self, inputs: Dict) -> List[float]:
         '''
         pred_labels: 0 / 1 (0: unsupported, 1: supported)
         max_support_probs: the probability of "supported" for the chunk that determin the final pred_label
@@ -26,8 +26,11 @@ class MiniCheck:
         support_prob_per_chunk: the probability of "supported" for each chunk
         '''
 
-        assert isinstance(docs, list) or isinstance(docs, np.ndarray), f"docs must be a list or np.ndarray, instead of {type(docs)};\n{docs}"
-        assert isinstance(claims, list) or isinstance(claims, np.ndarray), f"claims must be a list or np.ndarray, instead of {type(claims)};\n{claims}"  
+        docs = inputs['docs']
+        claims = inputs['claims']
+
+        assert isinstance(docs, list) or isinstance(docs, np.ndarray), f"docs must be a list or np.ndarray"
+        assert isinstance(claims, list) or isinstance(claims, np.ndarray), f"claims must be a list or np.ndarray"  
 
         max_support_prob, used_chunk, support_prob_per_chunk = self.model.fact_check(docs, claims)
         pred_label = [1 if prob > 0.5 else 0 for prob in max_support_prob]