MAINT Parameters validation for sklearn.datasets.load_breast_cancer (s…

…cikit-learn#26165) Co-authored-by: Jérémie du Boisberranger <[email protected]>
neurodata · Apr 14, 2023 · b58f448 · b58f448
1 parent c7949dd
commit b58f448
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/sklearn/datasets/_base.py b/sklearn/datasets/_base.py
@@ -670,6 +670,7 @@ def load_iris(*, return_X_y=False, as_frame=False):
  )
 
 
+@validate_params({"return_X_y": ["boolean"], "as_frame": ["boolean"]})
 def load_breast_cancer(*, return_X_y=False, as_frame=False):
  """Load and return the breast cancer wisconsin dataset (classification).
 

diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py
@@ -128,6 +128,7 @@ def _check_function_param_validation(
  "sklearn.datasets.fetch_lfw_people",
  "sklearn.datasets.fetch_olivetti_faces",
  "sklearn.datasets.fetch_rcv1",
+ "sklearn.datasets.load_breast_cancer",
  "sklearn.datasets.load_diabetes",
  "sklearn.datasets.load_iris",
  "sklearn.datasets.load_svmlight_file",