Replace sklearn estimator method _more_tags with __sklearn_tags__

folmos-at-orange · folmos-at-orange · commit fa8235984ef8 · 2024-12-10T09:34:53.000+01:00
This is an update to the estimator API which generates an error starting with scikit-learn 1.7. For details see scikit-learn/scikit-learn#28910
diff --git a/khiops/sklearn/estimators.py b/khiops/sklearn/estimators.py
@@ -268,8 +268,11 @@ def __init__(
         # Make sklearn get_params happy
         self.internal_sort = internal_sort
 
-    def _more_tags(self):
-        return {"allow_nan": True, "accept_large_sparse": False}
+    def __sklearn_tags__(self):
+        tags = super().__sklearn_tags__()
+        tags["allow_nan"] = True
+        tags["accept_large_sparse"] = False
+        return tags
 
     def _undefine_estimator_attributes(self):
         """Undefines all sklearn estimator attributes (ie. pass to "not fit" state)
@@ -1399,8 +1402,10 @@ def __init__(
                 )
             )
 
-    def _more_tags(self):
-        return {"require_y": True}
+    def __sklearn_tags__(self):
+        tags = super().__sklearn_tags__()
+        tags["require_y"] = True
+        return tags
 
     def _fit_check_dataset(self, ds):
         super()._fit_check_dataset(ds)
