MAINT Parameters validation for datasets.make_friedman3 (#25989)

Théophile Baranger · web-flow · commit f1491dff3b2a · 2023-03-29T11:16:13.000+02:00
diff --git a/sklearn/datasets/_samples_generator.py b/sklearn/datasets/_samples_generator.py
@@ -1169,6 +1169,13 @@ def make_friedman2(n_samples=100, *, noise=0.0, random_state=None):
     return X, y
 
 
+@validate_params(
+    {
+        "n_samples": [Interval(Integral, 1, None, closed="left")],
+        "noise": [Interval(Real, 0, None, closed="left")],
+        "random_state": ["random_state"],
+    }
+)
 def make_friedman3(n_samples=100, *, noise=0.0, random_state=None):
     """Generate the "Friedman #3" regression problem.
 
diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py
@@ -135,6 +135,7 @@ def _check_function_param_validation(
     "sklearn.datasets.make_classification",
     "sklearn.datasets.make_friedman1",
     "sklearn.datasets.make_friedman2",
+    "sklearn.datasets.make_friedman3",
     "sklearn.datasets.make_gaussian_quantiles",
     "sklearn.datasets.make_hastie_10_2",
     "sklearn.datasets.make_low_rank_matrix",
