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

Skip artifact-related tests if optuna version < v3.4

Browse Source
This commit is contained in:
porink0424
2024-06-05 10:58:14 +09:00
parent ac40496b32
commit 0de67317c1
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python_tests/test_api.py
+13 -2
View File
@@ -7,8 +7,6 @@ from unittest import TestCase
import optuna
from optuna import get_all_study_summaries
from optuna.artifacts import upload_artifact
from optuna.artifacts.exceptions import ArtifactNotFound
from optuna.study import StudyDirection
from optuna_dashboard._app import create_app
from optuna_dashboard._app import create_new_study
@@ -526,7 +524,14 @@ class APITestCase(TestCase):
self.assertEqual(status, 204)
self.assertEqual(len(get_all_study_summaries(storage)), 1)
@pytest.mark.skipif(
version.parse(optuna.__version__) < version.parse("3.4.0"),
reason="Needs optuna.artifacts",
)
def test_delete_study_with_removing_artifacts(self) -> None:
from optuna.artifacts import upload_artifact
from optuna.artifacts.exceptions import ArtifactNotFound
storage = optuna.storages.InMemoryStorage()
study = optuna.create_study(storage=storage)
with tempfile.TemporaryDirectory() as tmpdir_name:
@@ -556,7 +561,13 @@ class APITestCase(TestCase):
self.assertEqual(len(get_all_study_summaries(storage)), 0)
@pytest.mark.skipif(
version.parse(optuna.__version__) < version.parse("3.4.0"),
reason="Needs optuna.artifacts",
)
def test_delete_study_without_removing_artifacts(self) -> None:
from optuna.artifacts import upload_artifact
storage = optuna.storages.InMemoryStorage()
study = optuna.create_study(storage=storage)
with tempfile.TemporaryDirectory() as tmpdir_name: