MAINT Parameters validation for sklearn.utils.safe_mask (scikit-learn#26131)

Co-authored-by: jeremiedbb <jeremiedbb@yahoo.fr>

Author: elabongaatuo

sklearn/tests/test_public_functions.py

@@ -285,6 +285,7 @@ def _check_function_param_validation(
     "sklearn.utils.gen_batches",
     "sklearn.utils.graph.single_source_shortest_path_length",
     "sklearn.utils.resample",
+    "sklearn.utils.safe_mask",
 ]
 
 

sklearn/utils/__init__.py

@@ -112,6 +112,13 @@ def _in_unstable_openblas_configuration():
     return False
 
 
+@validate_params(
+    {
+        "X": ["array-like", "sparse matrix"],
+        "mask": ["array-like"],
+    },
+    prefer_skip_nested_validation=True,
+)
 def safe_mask(X, mask):
     """Return a mask which is safe to use on X.
 
@@ -120,7 +127,7 @@ def safe_mask(X, mask):
     X : {array-like, sparse matrix}
         Data on which to apply mask.
 
-    mask : ndarray
+    mask : array-like
         Mask to be used on X.
 
     Returns