Adapt to new default value.

Quantco · Feb 15, 2025 · 880bdee · 880bdee
1 parent 5ee607a
commit 880bdee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test__utils.py b/tests/test__utils.py
@@ -179,9 +179,9 @@ def test_convert_treatment_raise(use_pd):
     [
         (LinearRegression(), False),
         (
-            HistGradientBoostingClassifier(),
+            HistGradientBoostingClassifier(categorical_features=None),
             False,
-        ),  # The default for categorical_features will change to "from_dtype" in v1.6
+        ),
         (HistGradientBoostingClassifier(categorical_features="from_dtype"), True),
         (LGBMRegressor(), True),
         (XGBRegressor(), False),