From fa8a76d7b7faa2dc0da0a07b2f97b605947bbfe3 Mon Sep 17 00:00:00 2001
From: cfli <545999961@qq.com>
Date: Tue, 29 Oct 2024 20:27:36 +0800
Subject: [PATCH] add reranker score

---
 scripts/add_reranker_score.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/add_reranker_score.py b/scripts/add_reranker_score.py
index 61fafd7..af43dd9 100644
--- a/scripts/add_reranker_score.py
+++ b/scripts/add_reranker_score.py
@@ -134,6 +134,6 @@ if __name__ == '__main__':
             data[i]['neg_scores'].append(float(scores[score_idx]))
             score_idx += 1
 
-    with open(score_args.output_dir, 'w') as f:
+    with open(score_args.output_file, 'w') as f:
         for d in data:
             f.write(json.dumps(d) + '\n')
\ No newline at end of file