From 752443832aee943dca6f35b1882431e90d02c06c Mon Sep 17 00:00:00 2001 From: Jeonghwan Park <92216445+StableFluffy@users.noreply.github.com> Date: Fri, 28 Jun 2024 10:47:59 +0900 Subject: [PATCH] =?UTF-8?q?args=20=EC=97=90=EB=9F=AC=20=EC=88=98=EC=A0=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- evaluator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/evaluator.py b/evaluator.py index 96faac8..77bc09d 100644 --- a/evaluator.py +++ b/evaluator.py @@ -105,7 +105,7 @@ def process_item(client, row, judge_model, df_judge_template, output_file): f.write(json.dumps(row, ensure_ascii=False)) f.write('\n') -def process_file(client, file_path: Path, output_dir: Path, judge_model, df_judge_template, threads: int): +def process_file(client, file_path: Path, output_dir: Path, judge_model, df_judge_template, threads: int, args): print(f"- 현재 Processing : {file_path}") df_model_outputs = pd.read_json(file_path, lines=True) @@ -135,7 +135,7 @@ def main(): if output_file_path.exists(): print(f"이미 평가 완료.. : {file_path}") continue - process_file(client, file_path, output_dir, args.judge_model, df_judge_template, args.threads) + process_file(client, file_path, output_dir, args.judge_model, df_judge_template, args.threads, args) time.sleep(20) # ratelimit! if __name__ == "__main__":