MAINT Parameters validation for sklearn.model_selection.permutation_test_score (#26230)

Charlie-XIAO · web-flow · commit e20b3a66aad6 · 2023-04-25T14:29:55.000+02:00
diff --git a/sklearn/model_selection/_validation.py b/sklearn/model_selection/_validation.py
@@ -32,6 +32,7 @@
 from ..utils.metaestimators import _safe_split
 from ..utils._param_validation import (
     HasMethods,
+    Interval,
     Integral,
     StrOptions,
     validate_params,
@@ -1235,6 +1236,21 @@ def _check_is_permutation(indices, n_samples):
     return True
 
 
+@validate_params(
+    {
+        "estimator": [HasMethods("fit")],
+        "X": ["array-like", "sparse matrix"],
+        "y": ["array-like", None],
+        "groups": ["array-like", None],
+        "cv": ["cv_object"],
+        "n_permutations": [Interval(Integral, 1, None, closed="left")],
+        "n_jobs": [Integral, None],
+        "random_state": ["random_state"],
+        "verbose": ["verbose"],
+        "scoring": [StrOptions(set(get_scorer_names())), callable, None],
+        "fit_params": [dict, None],
+    }
+)
 def permutation_test_score(
     estimator,
     X,
diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py
@@ -247,6 +247,7 @@ def _check_function_param_validation(
     "sklearn.metrics.top_k_accuracy_score",
     "sklearn.metrics.zero_one_loss",
     "sklearn.model_selection.cross_validate",
+    "sklearn.model_selection.permutation_test_score",
     "sklearn.model_selection.train_test_split",
     "sklearn.neighbors.sort_graph_by_row_values",
     "sklearn.preprocessing.add_dummy_feature",
