From 0954154c166d86359273001f8f53e4260c554ee2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Th=C3=A9ophile=20Baranger?= Date: Tue, 28 Mar 2023 20:46:20 +0200 Subject: [PATCH] MAINT Parameters validation for datasets.make_sparse_uncorrelated --- sklearn/datasets/_samples_generator.py | 7 +++++++ sklearn/tests/test_public_functions.py | 1 + 2 files changed, 8 insertions(+) diff --git a/sklearn/datasets/_samples_generator.py b/sklearn/datasets/_samples_generator.py index 7b078eda40357..32b0b5c0b1719 100644 --- a/sklearn/datasets/_samples_generator.py +++ b/sklearn/datasets/_samples_generator.py @@ -1426,6 +1426,13 @@ def make_sparse_coded_signal( return map(np.squeeze, (Y, D, X)) +@validate_params( + { + "n_samples": [Interval(Integral, 1, None, closed="left")], + "n_features": [Interval(Integral, 1, None, closed="left")], + "random_state": ["random_state"], + } +) def make_sparse_uncorrelated(n_samples=100, n_features=10, *, random_state=None): """Generate a random regression problem with sparse uncorrelated design. diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py index d960639cee7fd..071e9bb33c622 100644 --- a/sklearn/tests/test_public_functions.py +++ b/sklearn/tests/test_public_functions.py @@ -141,6 +141,7 @@ def _check_function_param_validation( "sklearn.datasets.make_multilabel_classification", "sklearn.datasets.make_regression", "sklearn.datasets.make_sparse_coded_signal", + "sklearn.datasets.make_sparse_uncorrelated", "sklearn.decomposition.sparse_encode", "sklearn.feature_extraction.grid_to_graph", "sklearn.feature_extraction.img_to_graph",