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

Fix tests

Browse Source
This commit is contained in:
c-bata
2023-08-04 18:31:54 +09:00
parent 57ea65a93d
commit 2c3fb7711d
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python_tests/artifact/test_optuna_compatibility.py
+3 -3
View File
@@ -11,7 +11,7 @@ import pytest
@pytest.mark.skipif(
version.parse(optuna_ver) < version.Version("3.3.0"),
"Artifact is not implemented yet in Optuna",
reason="Artifact is not implemented yet in Optuna",
)
def test_list_optuna_artifacts() -> None:
from optuna.artifacts import FileSystemArtifactStore
@@ -33,8 +33,8 @@ def test_list_optuna_artifacts() -> None:
study.tell(trial, 0.0)
study_system_attrs = storage.get_study_system_attrs(study._study_id)
trial = storage.get_trial(trial._trial_id)
artifact_meta_list = list_trial_artifacts(study_system_attrs, trial)
frozen_trial = storage.get_trial(trial._trial_id)
artifact_meta_list = list_trial_artifacts(study_system_attrs, frozen_trial)
assert len(artifact_meta_list) == 1
artifact_id = artifact_meta_list[0]["artifact_id"]