wassname/optuna-dashboard
1
0
Fork 0
mirror of https://github.com/wassname/optuna-dashboard.git synced 2026-09-10 12:23:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert changes in upload_artifact api

Browse Source
This commit is contained in:
c-bata
2023-09-08 18:28:50 +09:00
parent 81d21eadab
commit 9dacaeece9
1 changed files with 4 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
optuna_dashboard/artifact/_backend.py
+4 -10
View File
@@ -166,7 +166,7 @@ def register_artifact_route(
def upload_artifact(
backend: ArtifactBackend,
study_or_trial: optuna.Trial | optuna.Study,
trial: optuna.Trial,
file_path: str,
*,
mimetype: Optional[str] = None,
@@ -202,6 +202,8 @@ def upload_artifact(
)
filename = os.path.basename(file_path)
storage = trial.storage
trial_id = trial._trial_id
artifact_id = str(uuid.uuid4())
guess_mimetype, guess_encoding = mimetypes.guess_type(filename)
artifact: ArtifactMeta = {
@@ -211,15 +213,7 @@ def upload_artifact(
"filename": filename,
}
attr_key = ARTIFACTS_ATTR_PREFIX + artifact_id
if isinstance(study_or_trial, optuna.Study):
storage = study_or_trial._storage
study_id = study_or_trial._study_id
storage.set_study_system_attr(study_id, attr_key, json.dumps(artifact))
else:
storage = study_or_trial.storage
trial_id = study_or_trial._trial_id
storage.set_trial_system_attr(trial_id, attr_key, json.dumps(artifact))
storage.set_trial_system_attr(trial_id, attr_key, json.dumps(artifact))
with open(file_path, "rb") as f:
backend.write(artifact_id, f)