updates

sktime · Jul 2, 2024 · 2451415 · 2451415
1 parent 6c17d12
commit 2451415
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/skpro/tests/test_create_container.py b/skpro/tests/test_create_container.py
@@ -5,14 +5,13 @@
 from sklearn.datasets import load_diabetes
 from sklearn.model_selection import train_test_split
 
-from skpro.utils.create_container import (
-    PandasAdapter,
-    PolarsAdapter,
-    get_config_adapter,
-)
+from skpro.utils.create_container import PandasAdapter, get_config_adapter
 from skpro.utils.set_output import check_transform_config
 from skpro.utils.validation._dependencies import _check_soft_dependencies
 
+if _check_soft_dependencies(["polars", "pyarrow"], severity="none"):
+    from skpro.utils.create_container import PolarsAdapter
+
 
 # test following cases, numpy input, polars input, pandas input
 @pytest.fixture