diff --git a/sklearn/datasets/_samples_generator.py b/sklearn/datasets/_samples_generator.py index 2352b5e50c6c5..d31ed5def2dd4 100644 --- a/sklearn/datasets/_samples_generator.py +++ b/sklearn/datasets/_samples_generator.py @@ -476,6 +476,12 @@ def sample_example(): return X, Y +@validate_params( + { + "n_samples": [Interval(Integral, 1, None, closed="left")], + "random_state": ["random_state"], + } +) def make_hastie_10_2(n_samples=12000, *, random_state=None): """Generate data for binary classification used in Hastie et al. 2009, Example 10.2. diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py index 9e1f95c4d057a..e6cf44efd1a27 100644 --- a/sklearn/tests/test_public_functions.py +++ b/sklearn/tests/test_public_functions.py @@ -133,6 +133,7 @@ def _check_function_param_validation( "sklearn.datasets.make_classification", "sklearn.datasets.make_friedman1", "sklearn.datasets.make_gaussian_quantiles", + "sklearn.datasets.make_hastie_10_2", "sklearn.datasets.make_low_rank_matrix", "sklearn.datasets.make_multilabel_classification", "sklearn.datasets.make_regression",