Fix unittest warning (#1802)

castorini · Mar 2, 2024 · bc5b3bf · bc5b3bf
1 parent 31b0087
commit bc5b3bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyserini/trectools/_base.py b/pyserini/trectools/_base.py
@@ -107,7 +107,7 @@ def reset_data(self):
         self.run_data = pd.DataFrame(columns=TrecRun.columns)
 
     def read_run(self, filepath: str, resort: bool = False) -> None:
-        self.run_data = pd.read_csv(filepath, sep='\s+', names=TrecRun.columns, dtype={'docid': 'str'})
+        self.run_data = pd.read_csv(filepath, sep='\s+', names=TrecRun.columns, dtype={'docid': 'str', 'score': 'float'})
         if resort:
             self.run_data.sort_values(["topic", "score"], inplace=True, ascending=[True, False])
             self.run_data["rank"] = self.run_data.groupby("topic")["score"].rank(ascending=False,method='first')