fix unit test

Author: sdpython

_unittests/ut_mlmodel/test_categories_to_integers.py

@@ -121,7 +121,8 @@ def test_categories_to_integers_grid_search(self):
                              LogisticRegression())
         self.assertRaise(lambda: test_sklearn_grid_search_cv(
             lambda: pipe, df), ValueError)
-        if compare_module_version(sklver, "0.24") >= 0:
+        if (compare_module_version(sklver, "0.24") >= 0 and
+                compare_module_version(pandas.__version__, "1.3") < 0):
             self.assertRaise(
                 lambda: test_sklearn_grid_search_cv(
                     lambda: pipe, X, y, categoriestointegers__single=[True, False]),

mlinsights/mlmodel/sklearn_testing.py

@@ -247,8 +247,8 @@ def test_sklearn_grid_search_cv(fct_model, X, y=None, sample_weight=None, **grid
     :raises:
         AssertionError
     """
-    X_train, y_train, w_train, X_test, y_test, w_test = train_test_split_with_none(
-        X, y, sample_weight)
+    X_train, y_train, w_train, X_test, y_test, w_test = (
+        train_test_split_with_none(X, y, sample_weight))
     model = fct_model()
     pipe = make_pipeline(model)
     name = model.__class__.__name__.lower()