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

optuna · Jun 5, 2024 · 0de6731 · 0de6731
1 parent ac40496
commit 0de6731
Showing 1 changed file with 13 additions and 2 deletions.
diff --git 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: