From fd0a54a437bfe897d3a87e64f4bfe82c174441a8 Mon Sep 17 00:00:00 2001 From: Jan Date: Sun, 31 Mar 2024 22:46:17 +0200 Subject: [PATCH] Ensure save_generations in generation only mode --- bigcode_eval/evaluator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bigcode_eval/evaluator.py b/bigcode_eval/evaluator.py index fa70000b9..c9b29a2f0 100644 --- a/bigcode_eval/evaluator.py +++ b/bigcode_eval/evaluator.py @@ -114,7 +114,7 @@ def save_json_files( save_generations_path: str, save_references_path: str, ) -> None: - if self.args.save_generations: + if self.args.save_generations or self.args.generation_only: with open(save_generations_path, "w") as fp: json.dump(generations, fp) print(f"generations were saved at {save_generations_path}")