Update test_probabilistic_models.py

unit8co · Dec 9, 2024 · a966b2c · a966b2c
1 parent a25f513
commit a966b2c
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/darts/tests/models/forecasting/test_probabilistic_models.py b/darts/tests/models/forecasting/test_probabilistic_models.py
@@ -16,6 +16,7 @@
     LightGBMModel,
     LinearRegressionModel,
     NotImportedModule,
+    StatsForecastAutoTBATS,
     XGBModel,
 )
 from darts.tests.conftest import TORCH_AVAILABLE, tfm_kwargs
@@ -92,6 +93,18 @@
         0.04,
         0.04,
     ),
+    (
+        StatsForecastAutoTBATS,
+        {
+            "season_length": 1,
+            "use_trend": False,
+            "use_damped_trend": False,
+            "use_boxcox": True,
+            "use_arma_errors": False,
+        },
+        0.04,
+        0.04,
+    ),
 ]
 
 xgb_test_params = {