diff --git a/sklearn/datasets/_base.py b/sklearn/datasets/_base.py index ed94d5ec1f0d7..abebe02b516dc 100644 --- a/sklearn/datasets/_base.py +++ b/sklearn/datasets/_base.py @@ -426,6 +426,12 @@ def load_descr(descr_file_name, *, descr_module=DESCR_MODULE): return fdescr +@validate_params( + { + "return_X_y": ["boolean"], + "as_frame": ["boolean"], + } +) def load_wine(*, return_X_y=False, as_frame=False): """Load and return the wine dataset (classification). diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py index d46ae07821ac2..105b2c80f1646 100644 --- a/sklearn/tests/test_public_functions.py +++ b/sklearn/tests/test_public_functions.py @@ -134,6 +134,7 @@ def _check_function_param_validation( "sklearn.datasets.load_iris", "sklearn.datasets.load_svmlight_file", "sklearn.datasets.load_svmlight_files", + "sklearn.datasets.load_wine", "sklearn.datasets.make_biclusters", "sklearn.datasets.make_blobs", "sklearn.datasets.make_checkerboard",