Skip to content

Commit

Permalink
Merge pull request #67 from DARPA-ASKEM/TS/optimize-output-file
Browse files Browse the repository at this point in the history
Adding .json results from dill for optimize
  • Loading branch information
Tom-Szendrey authored Mar 11, 2024
2 parents dfd1efd + 470e649 commit ef7f6e9
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions service/utils/tds.py
Original file line number Diff line number Diff line change
Expand Up @@ -213,6 +213,12 @@ def attach_files(output: dict, job_id, status="complete"):
results_filename = os.path.join(job_dir, "./optimize_results.dill")
results = output.get("OptResults", None)
if results is not None:
json_obj = json.loads(json.dumps(results, default=str))
json_filename = os.path.join(job_dir, "./optimize_result.json")
with open(json_filename, "w") as f:
json.dump(json_obj, f, ensure_ascii=False, indent=4)
files[json_filename] = "optimize_results.json"

with open(results_filename, "wb") as file:
dill.dump(results, file)
files[results_filename] = "optimize_results.dill"
Expand Down

0 comments on commit ef7f6e9

Please sign in to comment.