From d35874f6d9f32ab32a7b312e3750d01caa9f5519 Mon Sep 17 00:00:00 2001 From: Nathan Lambert Date: Mon, 12 Feb 2024 23:46:42 +0000 Subject: [PATCH] nit --- herm/utils.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/herm/utils.py b/herm/utils.py index d134085c..81599c03 100644 --- a/herm/utils.py +++ b/herm/utils.py @@ -40,8 +40,10 @@ def save_to_hub(results_dict: Dict, model_name: str, target_path: str, debug: bo dumped = json.dumps(results_dict, indent=4, sort_keys=True, default=str) if "scores" in target_path: scores_path = f"results/scores/{model_name}.json" + beaker_path = None else: scores_path = f"results/metrics/{model_name}.json" + beaker_path = "results/metrics.json" # save format for AI2 beaker to show results dirname = os.path.dirname(scores_path) os.makedirs(dirname, exist_ok=True) @@ -53,6 +55,11 @@ def save_to_hub(results_dict: Dict, model_name: str, target_path: str, debug: bo with open(scores_path, "w") as f: f.write(dumped) + # ai2 internal visualization, not needed external + if beaker_path: + with open(beaker_path, "w") as f: + f.write(dumped) + if not local_only: scores_url = api.upload_file( path_or_fileobj=scores_path,