Update rank_gpt.py

sunnweiwei · Feb 24, 2024 · 57ecaf0 · 57ecaf0
1 parent faae2dc
commit 57ecaf0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rank_gpt.py b/rank_gpt.py
@@ -277,7 +277,7 @@ def main():
     temp_file = tempfile.NamedTemporaryFile(delete=False).name
     from trec_eval import EvalFunction
 
-    EvalFunction.write_file(reranked_data, temp_file)
+    EvalFunction.write_file(new_results, temp_file)
     EvalFunction.main(THE_TOPICS[data], temp_file)