From d7e787759adae06e28be94d463634359a7deb111 Mon Sep 17 00:00:00 2001 From: Tom Szendrey Date: Thu, 14 Mar 2024 09:41:23 -0400 Subject: [PATCH] change the tensor to list before serializing --- 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 bbca5e2..ca71975 100644 --- a/service/utils/tds.py +++ b/service/utils/tds.py @@ -191,8 +191,16 @@ def attach_files(output: dict, job_id, status="complete"): files[output_filename] = "result.csv" risk_result = output.get("risk", None) + print("Tom") + print(risk_result) if risk_result is not None: - json_obj = json.loads(json.dumps(risk_result, default=str)) + # Update qoi (tensor) to a list before serializing with json.dumps + for k, v in risk_result.items(): + risk_result[k]["qoi"] = v["qoi"].tolist() + risk_json_obj = json.dumps(risk_result, default=str) + print(risk_json_obj) + json_obj = json.loads(risk_json_obj) + print(json_obj) json_filename = os.path.join(job_dir, "./risk.json") with open(json_filename, "w") as f: json.dump(json_obj, f, ensure_ascii=False, indent=4)