MAINT Parameters validation for datasets.make_sparse_spd_matrix (scikit-learn#26009)

Théophile Baranger · Itay · commit 68e038443c72 · 2023-04-15T23:06:53.000+03:00
diff --git a/sklearn/datasets/_samples_generator.py b/sklearn/datasets/_samples_generator.py
@@ -1537,6 +1537,16 @@ def make_spd_matrix(n_dim, *, random_state=None):
     return X
 
 
+@validate_params(
+    {
+        "dim": [Interval(Integral, 1, None, closed="left")],
+        "alpha": [Interval(Real, 0, 1, closed="both")],
+        "norm_diag": ["boolean"],
+        "smallest_coef": [Interval(Real, 0, 1, closed="both")],
+        "largest_coef": [Interval(Real, 0, 1, closed="both")],
+        "random_state": ["random_state"],
+    }
+)
 def make_sparse_spd_matrix(
     dim=1,
     *,
diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py
@@ -143,6 +143,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_spd_matrix",
     "sklearn.datasets.make_sparse_uncorrelated",
     "sklearn.datasets.make_spd_matrix",
     "sklearn.decomposition.sparse_encode",
