Skip to content

Commit

Permalink
fix(log_analysis): plot file extension
Browse files Browse the repository at this point in the history
  • Loading branch information
felixocker committed Aug 19, 2024
1 parent b916cef commit a73025f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/eval/math_eval/log_analysis.py
Original file line number Diff line number Diff line change
Expand Up @@ -799,9 +799,9 @@ def sanity_check_results(
print(f"{agent}: {mean_val:.4f} {relative_val:.4f} {values}")

if benchmark_type == "math":
img_name = "_".join(ln[:-3] for ln in log_names) + "_math_bench.png"
img_name = "_".join(ln[:-4] for ln in log_names) + "_math_bench.png"
elif benchmark_type in ("custom", "reduced"):
img_name = log_name[:-3] + ".png"
img_name = log_name[:-4] + ".png"
else:
raise ValueError(f"Unknown benchmark type `{benchmark_type}`.")
shutil.copy("math.eval.png", f"{log_folder}/{img_name}")

0 comments on commit a73025f

Please sign in to comment.