sklearn 1.6.1 compatibility and restrict <1.7

Author: davidbonet

hyperfast/hyperfast.py

@@ -146,6 +146,11 @@ def _get_tags(self) -> dict:
         tags["allow_nan"] = True
         return tags
 
+    def __sklearn_tags__(self):
+        tags = super().__sklearn_tags__()
+        tags.input_tags.allow_nan = True
+        return tags
+
     def _preprocess_fitting_data(
         self,
         x: np.ndarray | pd.DataFrame,

setup.cfg

@@ -19,6 +19,6 @@ python_requires = >=3.9
 install_requires =
     numpy>=1.26.2
     pandas>=2.1.3
-    scikit-learn>=1.3.2
+    scikit-learn>=1.3.2,<1.7
     torch>=1.10.2
     requests>=2.31.0