Update test_metrics.py

unit8co · Jan 22, 2025 · bf8150f · bf8150f
1 parent dbf4fc2
commit bf8150f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/darts/tests/metrics/test_metrics.py b/darts/tests/metrics/test_metrics.py
@@ -227,6 +227,7 @@ def test_ope_zero(self):
             (metrics.msse, False, {}),
             (metrics.rmsse, False, {}),
             (metrics.mape, False, {}),
+            (metrics.wmape, False, {}),
             (metrics.smape, False, {}),
             (metrics.ope, False, {}),
             (metrics.marre, False, {}),
@@ -819,6 +820,7 @@ def test_output_type_time_dependent(self, config):
                 (metrics.msse, False),
                 (metrics.rmsse, False),
                 (metrics.mape, False),
+                (metrics.wmape, False),
                 (metrics.smape, False),
                 (metrics.ope, False),
                 (metrics.marre, False),
@@ -921,6 +923,7 @@ def test_reduction_fn_validity(self, config):
             (metrics.msse, 0, False, {}),
             (metrics.rmsse, 0, False, {}),
             (metrics.mape, 0, False, {}),
+            (metrics.wmape, 0, False, {}),
             (metrics.smape, 0, False, {}),
             (metrics.ope, 0, False, {}),
             (metrics.marre, 0, False, {}),
@@ -1368,6 +1371,7 @@ def test_multiple_ts_rmse(self):
             (metrics.rmse, "max", {}),
             (metrics.rmsle, "max", {}),
             (metrics.mape, "max", {}),
+            (metrics.wmape, "max", {}),
             (metrics.smape, "max", {}),
             (metrics.ope, "max", {}),
             (metrics.marre, "max", {}),
@@ -1662,6 +1666,7 @@ def helper_test_non_aggregate(self, metric, is_aggregate, val_exp=None):
                     metrics.msse,
                     metrics.rmsse,
                     metrics.mape,
+                    metrics.wmape,
                     metrics.smape,
                     metrics.ope,
                     metrics.marre,