From 24514156d63c899551d19d7850d9e81cb5817ffd Mon Sep 17 00:00:00 2001 From: julian fong Date: Tue, 2 Jul 2024 08:42:31 -0400 Subject: [PATCH] updates --- skpro/tests/test_create_container.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/skpro/tests/test_create_container.py b/skpro/tests/test_create_container.py index 3d06c178b..e78bd0883 100644 --- 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