From a966b2c92bd6f78f025ddd857b307f55c30554bb Mon Sep 17 00:00:00 2001
From: he weilin <heweilin19971212@163.com>
Date: Mon, 9 Dec 2024 17:16:21 +0800
Subject: [PATCH] Update test_probabilistic_models.py

---
 .../models/forecasting/test_probabilistic_models.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/darts/tests/models/forecasting/test_probabilistic_models.py b/darts/tests/models/forecasting/test_probabilistic_models.py
index 141fd43dcd..3b4352aad1 100644
--- 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 = {