Implement estimator__sklearn_tags__ method

folmos-at-orange · folmos-at-orange · commit 68c665645131 · 2024-12-10T10:19:41.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,6 +268,11 @@ def __init__(
         # Make sklearn get_params happy
         self.internal_sort = internal_sort
 
+    def __sklearn_tags__(self):
+        tags = super().__sklearn_tags__()
+        for tag, value in self._more_tags():
+            setattr(tags, tag, value)
+
     def _more_tags(self):
         return {"allow_nan": True, "accept_large_sparse": False}
 
