From 0de67317c1d6f440b372b53fe5f910f0af2f52bd Mon Sep 17 00:00:00 2001 From: porink0424 Date: Wed, 5 Jun 2024 10:58:14 +0900 Subject: [PATCH] Skip artifact-related tests if optuna version < v3.4 --- python_tests/test_api.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/python_tests/test_api.py b/python_tests/test_api.py index 129bf0ae..6d6b960a 100644 --- a/python_tests/test_api.py +++ b/python_tests/test_api.py @@ -7,8 +7,6 @@ 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 @@ def test_delete_study(self) -> None: 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 @@ def test_delete_study_with_removing_artifacts(self) -> None: 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: