8000 MAINT Parameters validation for sklearn.datasets.load_iris (#26177) · scikit-learn/scikit-learn@0f295aa · GitHub
[go: up one dir, main page]
Skip to content

Commit 0f295aa

Browse files
dvignesh1995dvignesh1995
authored
MAINT Parameters validation for sklearn.datasets.load_iris (#26177)
1 parent 5fbd02e commit 0f295aa

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

‎sklearn/datasets/_base.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
from ..utils import check_random_state
2222
from ..utils import check_pandas_support
2323
from ..utils.fixes import _open_binary, _open_text, _read_text, _contents
24+
from ..utils._param_validation import validate_params
2425

2526
import numpy as np
2627

@@ -545,6 +546,7 @@ def load_wine(*, return_X_y=False, as_frame=False):
545546
)
546547

547548

549+
@validate_params({"return_X_y": ["boolean"], "as_frame": ["boolean"]})
548550
def load_iris(*, return_X_y=False, as_frame=False):
549551
"""Load and return the iris dataset (classification).
550552

‎sklearn/tests/test_public_functions.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,7 @@ def _check_function_param_validation(
128128
"sklearn.datasets.fetch_lfw_people",
129129
"sklearn.datasets.fetch_olivetti_faces",
130130
"sklearn.datasets.fetch_rcv1",
131+
"sklearn.datasets.load_iris",
131132
"sklearn.datasets.load_svmlight_file",
132133
"sklearn.datasets.load_svmlight_files",
133134
"sklearn.datasets.make_biclusters",

0 commit comments

Comments
 (0)
0