MAINT Parameters validation for sklearn.datasets.load_linnerud (sciki…

…t-learn#26199)
neurodata · Apr 19, 2023 · 1c04a37 · 1c04a37
1 parent 9cddedf
commit 1c04a37
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/sklearn/datasets/_base.py b/sklearn/datasets/_base.py
@@ -1067,6 +1067,12 @@ def load_diabetes(*, return_X_y=False, as_frame=False, scaled=True):
  )
 
 
+@validate_params(
+ {
+ "return_X_y": ["boolean"],
+ "as_frame": ["boolean"],
+ }
+)
 def load_linnerud(*, return_X_y=False, as_frame=False):
  """Load and return the physical exercise Linnerud dataset.
 

diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py
@@ -133,6 +133,7 @@ def _check_function_param_validation(
  "sklearn.datasets.load_breast_cancer",
  "sklearn.datasets.load_diabetes",
  "sklearn.datasets.load_iris",
+ "sklearn.datasets.load_linnerud",
  "sklearn.datasets.load_svmlight_file",
  "sklearn.datasets.load_svmlight_files",
  "sklearn.datasets.make_biclusters",