Update run_evaluation.py

sunnweiwei · Feb 24, 2024 · faae2dc · faae2dc
1 parent 081885e
commit faae2dc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/run_evaluation.py b/run_evaluation.py
@@ -95,7 +95,7 @@
 
     # Create an empty text file to write results, and pass the name to eval
     temp_file = tempfile.NamedTemporaryFile(delete=False).name
-    EvalFunction.write_file(reranked_data, temp_file)
+    EvalFunction.write_file(new_results, temp_file)
     EvalFunction.main(THE_TOPICS[data], temp_file)
 
 
@@ -127,5 +127,5 @@
     from trec_eval import EvalFunction
 
     temp_file = tempfile.NamedTemporaryFile(delete=False).name
-    EvalFunction.write_file(reranked_data, temp_file)
+    EvalFunction.write_file(new_results, temp_file)
     EvalFunction.main(THE_TOPICS[data], temp_file)