Merge pull request #713 from c-bata/fix-upload-trial-artifact

Fix a minor bug that `upload_artifact` api does not return the latest artifacts.
optuna · Nov 28, 2023 · 520d4d3 · 520d4d3
2 parents 4aa4517 + 265dfb4
commit 520d4d3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/optuna_dashboard/artifact/_backend.py b/optuna_dashboard/artifact/_backend.py
@@ -139,6 +139,7 @@ def upload_trial_artifact_api(study_id: int, trial_id: int) -> dict[str, Any]:
         storage.set_trial_system_attr(trial_id, attr_key, json.dumps(artifact))
         response.status = 201
 
+        trial = storage.get_trial(trial_id)  # Fetch trial.system_attrs again.
         return {
             "artifact_id": artifact_id,
             "artifacts": list_trial_artifacts(storage.get_study_system_attrs(study_id), trial),