MAINT Parameters validation for sklearn.datasets.load_iris (#26177)

dvignesh1995 · web-flow · commit 0f295aa35bdb · 2023-04-14T10:27:10.000+02:00
diff --git 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
 
@@ -545,6 +546,7 @@ def load_wine(*, return_X_y=False, as_frame=False):
     )
 
 
+@validate_params({"return_X_y": ["boolean"], "as_frame": ["boolean"]})
 def load_iris(*, return_X_y=False, as_frame=False):
     """Load and return the iris 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_iris",
     "sklearn.datasets.load_svmlight_file",
     "sklearn.datasets.load_svmlight_files",
     "sklearn.datasets.make_biclusters",
