MAINT Parameters validation for sklearn.covariance.ledoit_wolf_shrinkage (#26200)

ROMEEZHOU · web-flow · commit 9cddedfc8675 · 2023-04-19T16:40:29.000+02:00
diff --git a/sklearn/covariance/_shrunk_covariance.py b/sklearn/covariance/_shrunk_covariance.py
@@ -272,6 +272,13 @@ def fit(self, X, y=None):
 # Ledoit-Wolf estimator
 
 
+@validate_params(
+    {
+        "X": ["array-like"],
+        "assume_centered": ["boolean"],
+        "block_size": [Interval(Integral, 1, None, closed="left")],
+    }
+)
 def ledoit_wolf_shrinkage(X, assume_centered=False, block_size=1000):
     """Estimate the shrunk Ledoit-Wolf covariance matrix.
 
diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py
@@ -117,6 +117,7 @@ def _check_function_param_validation(
     "sklearn.cluster.cluster_optics_xi",
     "sklearn.cluster.ward_tree",
     "sklearn.covariance.empirical_covariance",
+    "sklearn.covariance.ledoit_wolf_shrinkage",
     "sklearn.covariance.shrunk_covariance",
     "sklearn.datasets.dump_svmlight_file",
     "sklearn.datasets.fetch_20newsgroups",
