From dc56226bf899df7e19bd51428387210e1804c5b1 Mon Sep 17 00:00:00 2001 From: Alex Barron Date: Fri, 6 Dec 2024 15:54:29 -0800 Subject: [PATCH] only write top level results --- llms/mlx_lm/evaluate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llms/mlx_lm/evaluate.py b/llms/mlx_lm/evaluate.py index cbb7917a..0efae2fe 100644 --- a/llms/mlx_lm/evaluate.py +++ b/llms/mlx_lm/evaluate.py @@ -342,7 +342,7 @@ def main(): filename = f"eval_{args.model.replace('/', '_')}_{('_'.join(args.tasks))}_{args.num_shots:02d}.json" output_path = Path(args.output_dir) / filename - output_path.write_text(json.dumps(results, indent=4)) + output_path.write_text(json.dumps(results["results"], indent=4)) print("Results:") for result in results["results"].values(): print(json.dumps(result, indent=4))