Implement estimator__sklearn_tags__ method

folmos-at-orange · folmos-at-orange · commit cb319a2a9bfe · 2024-12-10T12:17:00.000+01:00
This is an update to the estimator API which generates an error starting with scikit-learn 1.7. We keep `_more_tags` for backward compatibility and implement `__sklearn_tags__` with it. For details see scikit-learn/scikit-learn#28910
diff --git a/khiops/sklearn/estimators.py b/khiops/sklearn/estimators.py
@@ -268,8 +268,24 @@ def __init__(
         # Make sklearn get_params happy
         self.internal_sort = internal_sort
 
+    def __sklearn_tags__(self):
+        # Error if BaseEstimator does not support __sklearn_tags__
+        if not hasattr(BaseEstimator, "__sklearn_tags__"):
+            raise AttributeError("__sklearn_tags__ API unsupported.")
+
+        # Set the tags from _more_tags
+        tags = super().__sklearn_tags__()
+        for tag, tag_value in self._more_tags():
+            if tag == "allow_nan":
+                tags.input_tags.allow_nan = tag_value
+            elif tag == "requires_y":
+                tags.target_tag.required = tag_value
+            elif tag == "preserves_dtype":
+                tags.transformer_tags.preserves_dtype = tag_value
+        return tags
+
     def _more_tags(self):
-        return {"allow_nan": True, "accept_large_sparse": False}
+        return {"allow_nan": True}
 
     def _undefine_estimator_attributes(self):
         """Undefines all sklearn estimator attributes (ie. pass to "not fit" state)
@@ -1400,7 +1416,7 @@ def __init__(
             )
 
     def _more_tags(self):
-        return {"require_y": True}
+        return {"requires_y": True}
 
     def _fit_check_dataset(self, ds):
         super()._fit_check_dataset(ds)
@@ -2783,7 +2799,7 @@ def __init__(
         self.keep_initial_variables = keep_initial_variables
         self._khiops_model_prefix = "R_"
 
-    def more_tags(self):
+    def _more_tags(self):
         return {"preserves_dtype": []}
 
     def _categorical_transform_method(self):
