MAINT Parameters validation for sklearn.metrics.d2_tweedie_score (scikit-learn#25975)

A-H-Mansoury · Itay · commit 611858b865a9 · 2023-04-15T23:06:52.000+03:00
diff --git a/sklearn/metrics/_regression.py b/sklearn/metrics/_regression.py
@@ -1238,6 +1238,17 @@ def mean_gamma_deviance(y_true, y_pred, *, sample_weight=None):
     return mean_tweedie_deviance(y_true, y_pred, sample_weight=sample_weight, power=2)
 
 
+@validate_params(
+    {
+        "y_true": ["array-like"],
+        "y_pred": ["array-like"],
+        "sample_weight": ["array-like", None],
+        "power": [
+            Interval(Real, None, 0, closed="right"),
+            Interval(Real, 1, None, closed="left"),
+        ],
+    }
+)
 def d2_tweedie_score(y_true, y_pred, *, sample_weight=None, power=0):
     """D^2 regression score function, fraction of Tweedie deviance explained.
 
@@ -1257,7 +1268,7 @@ def d2_tweedie_score(y_true, y_pred, *, sample_weight=None, power=0):
     y_pred : array-like of shape (n_samples,)
         Estimated target values.
 
-    sample_weight : array-like of shape (n_samples,), optional
+    sample_weight : array-like of shape (n_samples,), default=None
         Sample weights.
 
     power : float, default=0
diff --git a/sklearn/tests/test_public_functions.py b/sklearn/tests/test_public_functions.py
@@ -162,6 +162,7 @@ def _check_function_param_validation(
     "sklearn.metrics.confusion_matrix",
     "sklearn.metrics.coverage_error",
     "sklearn.metrics.d2_pinball_score",
+    "sklearn.metrics.d2_tweedie_score",
     "sklearn.metrics.dcg_score",
     "sklearn.metrics.det_curve",
     "sklearn.metrics.f1_score",
