ut

Author: sdpython

.travis.yml

@@ -7,7 +7,7 @@ matrix:
   - python: 3.9
     name: "Py39-024+"
     env:
-      - sklver=">=0.24.0"
+      - sklver=">=0.24.2"
       - jlver=">=1.0"
   - python: 3.8
     name: "Py38-023"

_unittests/ut_mlmodel/test_categories_to_integers.py

@@ -119,10 +119,11 @@ def test_categories_to_integers_grid_search(self):
                              LogisticRegression())
         self.assertRaise(lambda: test_sklearn_grid_search_cv(
             lambda: pipe, df), ValueError)
-        self.assertRaise(
-            lambda: test_sklearn_grid_search_cv(
-                lambda: pipe, X, y, categoriestointegers__single=[True, False]),
-            ValueError, "Unable to find category value")
+        if compare_module_version(sklver, "0.24") >= 0:
+            self.assertRaise(
+                lambda: test_sklearn_grid_search_cv(
+                    lambda: pipe, X, y, categoriestointegers__single=[True, False]),
+                ValueError, "Unable to find category value")
         pipe = make_pipeline(CategoriesToIntegers(),
                              Imputer(strategy='most_frequent'),
                              LogisticRegression(n_jobs=1))