From f6f127e93de44e8286e8ec16b5ec5d5ffa78814c Mon Sep 17 00:00:00 2001 From: Daniel Chang Date: Tue, 7 May 2024 12:50:16 -0400 Subject: [PATCH] observable sanity check --- service/utils/tds.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/service/utils/tds.py b/service/utils/tds.py index 2d078e2..27f9f95 100644 --- a/service/utils/tds.py +++ b/service/utils/tds.py @@ -114,7 +114,15 @@ def fetch_model(model_config_id, job_id): amr_path = os.path.join(job_dir, f"./{model_config_id}.json") with open(amr_path, "w") as file: - json.dump(model_response.json()["configuration"], file) + # Ensure we don't have null observables which can be problematic downstream, if so convert + # to empty list + model_json = model_response.json()["configuration"] + if "semantics" in model_json and "ode" in model_json["semantics"]: + ode = model_json["semantics"]["ode"] + if "observables" in ode and ode["observables"] is None: + ode["observables"] = [] + + json.dump(model_json, file) return amr_path