Skip to content

Commit 9eba1ce

Browse files
committed
skiping polars transformation function tests if polars is not installed
1 parent c07ee94 commit 9eba1ce

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

python/tests/engine/test_python.py

+8
Original file line numberDiff line numberDiff line change
@@ -3349,6 +3349,10 @@ def plus_one(col1):
33493349
assert result["plus_one_tf_name_"][0] == 2
33503350
assert result["plus_one_tf_name_"][1] == 3
33513351

3352+
@pytest.mark.skipif(
3353+
not HAS_POLARS,
3354+
reason="Polars is not installed.",
3355+
)
33523356
def test_apply_transformation_function_polars_udf_python_mode(self, mocker):
33533357
# Arrange
33543358
mocker.patch("hopsworks_common.client.get_instance")
@@ -3390,6 +3394,10 @@ def plus_one(col1):
33903394
assert result["plus_one_tf_name_"][0] == 2
33913395
assert result["plus_one_tf_name_"][1] == 3
33923396

3397+
@pytest.mark.skipif(
3398+
not HAS_POLARS,
3399+
reason="Polars is not installed.",
3400+
)
33933401
def test_apply_transformation_function_polars_udf_pandas_mode(self, mocker):
33943402
# Arrange
33953403
mocker.patch("hopsworks_common.client.get_instance")

0 commit comments

Comments
 (0)