replacing deprecated "squared": False from mean_squared_error (#2554)

unit8co · Nov 7, 2024 · a674ed2 · a674ed2
1 parent 36d041c
commit a674ed2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/darts/tests/metrics/test_metrics.py b/darts/tests/metrics/test_metrics.py
@@ -43,7 +43,7 @@ def metric_cov(y_true, y_pred, **kwargs):
     y_pred = y_pred[:, 0]
     return (
         100.0
-        * sklearn.metrics.mean_squared_error(y_true, y_pred, squared=False)
+        * sklearn.metrics.root_mean_squared_error(y_true, y_pred)
         / np.mean(y_true)
     )
 
@@ -1400,7 +1400,7 @@ def test_multiple_ts(self, config):
             (metrics.merr, metric_residuals, {}, {}),
             (metrics.mae, sklearn.metrics.mean_absolute_error, {}, {}),
             (metrics.mse, sklearn.metrics.mean_squared_error, {}, {}),
-            (metrics.rmse, sklearn.metrics.mean_squared_error, {"squared": False}, {}),
+            (metrics.rmse, sklearn.metrics.root_mean_squared_error, {}, {}),
             (metrics.rmsle, metric_rmsle, {}, {}),
             (metrics.mape, sklearn_mape, {}, {}),
             (metrics.smape, metric_smape, {}, {}),