revert test import

HideakiImamura · Jul 11, 2024 · 12a023a · 12a023a
1 parent 12d9006
commit 12a023a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/tests/artifacts_tests/test_backoff.py b/tests/artifacts_tests/test_backoff.py
@@ -2,8 +2,9 @@
 import uuid
 
 from optuna.artifacts import Backoff
-from tests.artifacts_tests.stubs import FailArtifactStore
-from tests.artifacts_tests.stubs import InMemoryArtifactStore
+
+from .stubs import FailArtifactStore
+from .stubs import InMemoryArtifactStore
 
 
 def test_backoff_time() -> None:

diff --git a/tests/storages_tests/rdb_tests/test_storage.py b/tests/storages_tests/rdb_tests/test_storage.py
@@ -31,9 +31,10 @@
 from optuna.testing.tempfile_pool import NamedTemporaryFilePool
 from optuna.trial import FrozenTrial
 from optuna.trial import TrialState
-from tests.storages_tests.rdb_tests.create_db import mo_objective_test_upgrade
-from tests.storages_tests.rdb_tests.create_db import objective_test_upgrade
-from tests.storages_tests.rdb_tests.create_db import objective_test_upgrade_distributions
+
+from .create_db import mo_objective_test_upgrade
+from .create_db import objective_test_upgrade
+from .create_db import objective_test_upgrade_distributions
 
 
 def test_init() -> None: