MAINT Parameters validation for sklearn.datasets.load_wine (#26196)

jiawei-zhang-a · web-flow · commit 4df6c646e5d4 · 2023-04-19T16:52:38.000+02:00
diff --git 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
@@ -136,6 +136,7 @@ def _check_function_param_validation(
     "sklearn.datasets.load_linnerud",
     "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",
