Skip to content

Commit 722e645

Browse files
authored
fix probabilistic tests for prophet (#2815)
1 parent 4a84d93 commit 722e645

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

darts/tests/models/forecasting/test_probabilistic_models.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@
6767

6868
lgbm_available = not isinstance(LightGBMModel, NotImportedModule)
6969
cb_available = not isinstance(CatBoostModel, NotImportedModule)
70+
prophet_available = not isinstance(Prophet, NotImportedModule)
7071

7172
# conformal models require a fitted base model
7273
# in tests below, the model is re-trained for new input series.
@@ -238,7 +239,6 @@
238239
extra_configs = [
239240
(ExponentialSmoothing, {"random_state": 42}, 0.3, None),
240241
(VARIMA, {"random_state": 42}, 0.03, None),
241-
(Prophet, {"random_state": 42}, 0.03, None),
242242
(KalmanForecaster, {"random_state": 42}, 0.03, None),
243243
(
244244
LinearRegressionModel,
@@ -346,6 +346,8 @@
346346
None,
347347
),
348348
]
349+
if prophet_available:
350+
extra_configs += [(Prophet, {"random_state": 42}, 0.03, None)]
349351

350352

351353
@pytest.mark.slow

0 commit comments

Comments
 (0)