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)