fix pd categorical test

oegedijk · Mar 18, 2024 · c775bbe · c775bbe
1 parent d28d18d
commit c775bbe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)