Skip to content
This repository has been archived by the owner on Oct 17, 2024. It is now read-only.

Commit

Permalink
args 에러 수정
Browse files Browse the repository at this point in the history
  • Loading branch information
StableFluffy authored Jun 28, 2024
1 parent d0d974f commit 7524438
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions evaluator.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down Expand Up @@ -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__":
Expand Down

0 comments on commit 7524438

Please sign in to comment.