From c775bbec43896226e89311e71f151daf0b4cdafb Mon Sep 17 00:00:00 2001 From: Oege Dijk Date: Mon, 18 Mar 2024 21:27:48 +0100 Subject: [PATCH] fix pd categorical test --- tests/test_regression_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_regression_base.py b/tests/test_regression_base.py index 03718af..9226576 100644 --- a/tests/test_regression_base.py +++ b/tests/test_regression_base.py @@ -76,7 +76,7 @@ def test_get_col(precalculated_rf_regression_explainer): precalculated_rf_regression_explainer.get_col("Gender"), pd.Series ) assert isinstance( - precalculated_rf_regression_explainer.get_col("Gender"), pd.CategoricalDtype() + precalculated_rf_regression_explainer.get_col("Gender"), pd.CategoricalDtype ) assert isinstance(precalculated_rf_regression_explainer.get_col("Age"), pd.Series)