From ab2fa6f447aad09563eb4a2c0b047ef4f4ad21d2 Mon Sep 17 00:00:00 2001 From: "Thomas J. Fan" Date: Mon, 4 Dec 2023 17:22:59 -0500 Subject: [PATCH] CLN Update docs and test name for polars output in ColumnTransformer --- sklearn/utils/tests/test_validation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sklearn/utils/tests/test_validation.py b/sklearn/utils/tests/test_validation.py index 5f08244610cea..1f847dbd55d62 100644 --- a/sklearn/utils/tests/test_validation.py +++ b/sklearn/utils/tests/test_validation.py @@ -1775,7 +1775,7 @@ def test_is_polars_df_other_libraries(constructor_name, minversion): assert _is_polars_df(df) -def test_is_polars_df_pandas_not_installed(): +def test_is_polars_df_for_duck_typed_polars_dataframe(): """Check _is_polars_df for object that looks like a polars dataframe""" class NotAPolarsDataFrame: @@ -1992,7 +1992,7 @@ def test_check_array_multiple_extensions( def test_num_samples_dataframe_protocol(): - """Use DataFrame protocol to get n_samples from polars dataframe.""" + """Use the DataFrame interchange protocol to get n_samples from polars.""" pl = pytest.importorskip("polars") df = pl.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6]})