From c4c79c1d2a94e155d16259e6108e7ce191a9bb81 Mon Sep 17 00:00:00 2001 From: "Lennart J. Kurzweg (Nx2)" Date: Sat, 5 Oct 2024 21:08:31 +0200 Subject: [PATCH] prevent visualize from crashing If somehting exists in saved_results.json, that doesnt in the suite_settings (comented out e.g.) it crashed. Now it doesnt. --- visualize.py | 23 +++++++++++++---------- 1 file changed, 13 insertions(+), 10 deletions(-) diff --git a/visualize.py b/visualize.py index 2dd3b87..1e6db47 100644 --- a/visualize.py +++ b/visualize.py @@ -18,16 +18,19 @@ def get_df() -> pd.DataFrame: data = json.load(f) raw_data = [] for test_hash, test_data in data.items(): - raw_data.append({ - "hash": test_hash, - "model_name": models[test_data['model_id']].display_name, - "model_size": models[test_data['model_id']].parameter_count_in_b, - "technique_name": techniques[test_data['technique_id']].name, - "model_technique": f"{models[test_data['model_id']].display_name}:{ techniques[test_data['technique_id']].name}", - "seed": test_data['seed'], - "test_name": tests[test_data['test_id']].name, - "validation": test_data['validation'] - }) + try: + raw_data.append({ + "hash": test_hash, + "model_name": models[test_data['model_id']].display_name, + "model_size": models[test_data['model_id']].parameter_count_in_b, + "technique_name": techniques[test_data['technique_id']].name, + "model_technique": f"{models[test_data['model_id']].display_name}:{ techniques[test_data['technique_id']].name}", + "seed": test_data['seed'], + "test_name": tests[test_data['test_id']].name, + "validation": test_data['validation'] + }) + except KeyError: + pass df = pd.DataFrame(raw_data)