From 4629496f94d9a964e3276f4e3b927af39c598533 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Th=C3=A9ophile=20Baranger?= Date: Fri, 24 Mar 2023 19:51:08 +0100 Subject: [PATCH] MAINT Parameters validation for datasets.make_hastie_10_2 --- sklearn/datasets/_samples_generator.py | 6 ++++++ sklearn/tests/test_public_functions.py | 1 + 2 files changed, 7 insertions(+) diff --git a/sklearn/datasets/_samples_generator.py b/sklearn/datasets/_samples_generator.py index 192f8e6759ddb..73afa0d63c876 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 4b1934c378fbf..bd7503672a4e9 100644 --- a/sklearn/tests/test_public_functions.py +++ b/sklearn/tests/test_public_functions.py @@ -132,6 +132,7 @@ def _check_function_param_validation( "sklearn.datasets.make_circles", "sklearn.datasets.make_classification", "sklearn.datasets.make_friedman1", + "sklearn.datasets.make_hastie_10_2", "sklearn.datasets.make_low_rank_matrix", "sklearn.datasets.make_multilabel_classification", "sklearn.datasets.make_regression",