From fd5670905622883dd55563c69f4020ff22739598 Mon Sep 17 00:00:00 2001 From: Charlie-XIAO Date: Thu, 13 Apr 2023 02:59:15 +0800 Subject: [PATCH] MAINT Parameters validation for sklearn.datasets.load_breast_cancer --- sklearn/datasets/_base.py | 2 ++ sklearn/tests/test_public_functions.py | 1 + 2 files changed, 3 insertions(+) diff --git a/sklearn/datasets/_base.py b/sklearn/datasets/_base.py index 6331582cffb4a..c1e5b481d0c49 100644 --- a/sklearn/datasets/_base.py +++ b/sklearn/datasets/_base.py @@ -21,6 +21,7 @@ from ..utils import check_random_state from ..utils import check_pandas_support from ..utils.fixes import _open_binary, _open_text, _read_text, _contents +from ..utils._param_validation import validate_params import numpy as np @@ -668,6 +669,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 index 62513b0f63cce..2310cf2e49ef5 100644 --- 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_svmlight_file", "sklearn.datasets.load_svmlight_files", "sklearn.datasets.make_biclusters",