From 4a036c59b57199b8bf07218c8428c4e5a1a49239 Mon Sep 17 00:00:00 2001 From: Daniel Chang Date: Mon, 5 Feb 2024 13:38:40 -0500 Subject: [PATCH] iso date format --- service/utils/tds.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/service/utils/tds.py b/service/utils/tds.py index 85f4603..3735c52 100644 --- a/service/utils/tds.py +++ b/service/utils/tds.py @@ -68,9 +68,9 @@ def update_tds_status(job_id, status, result_files=[], start=False, finish=False tds_payload = tds_payload.json() if start: - tds_payload["start_time"] = datetime.now() + tds_payload["start_time"] = datetime.now().isoformat() if finish: - tds_payload["completed_time"] = datetime.now() + tds_payload["completed_time"] = datetime.now().isoformat() tds_payload["status"] = status if result_files: @@ -141,6 +141,7 @@ def fetch_dataset(dataset: dict, job_id): if response.status_code >= 300: raise HTTPException(status_code=400, detail="Unable to retrieve dataset") + df = pd.read_csv(response.json()["url"]) df = df.rename(columns=dataset["mappings"]) @@ -165,6 +166,7 @@ def fetch_inferred_parameters(parameters_id: Optional[str], job_id): job_dir = get_job_dir(job_id) logging.debug(f"Fetching inferred parameters {parameters_id}") download_url = f"{TDS_URL}{TDS_SIMULATIONS}/{parameters_id}/download-url?filename=parameters.dill" + parameters_url = tds_session().get(download_url).json()["url"] # response = tds_session().get(parameters_url)