pmdarima to 2.0.4

h2oai · May 22, 2024 · 1439a0f · 1439a0f
1 parent 289d97f
commit 1439a0f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/transformers/timeseries/auto_arima_forecast.py b/transformers/timeseries/auto_arima_forecast.py
@@ -16,7 +16,7 @@
 class MyAutoArimaTransformer(CustomTimeSeriesTransformer):
     _binary = False
     _multiclass = False
-    _modules_needed_by_name = ['pmdarima==1.8.3']
+    _modules_needed_by_name = ['pmdarima==2.0.4']
     _included_model_classes = None
     _testing_can_skip_failure = False  # ensure tested as if shouldn't fail
     _lag_recipe_allowed = True

diff --git a/transformers/timeseries/parallel_auto_arima_forecast.py b/transformers/timeseries/parallel_auto_arima_forecast.py
@@ -52,7 +52,7 @@ class MyParallelAutoArimaTransformer(CustomTimeSeriesTransformer):
     """Implementation of the ARIMA transformer using a pool of processes to fit models in parallel"""
     _binary = False
     _multiclass = False
-    _modules_needed_by_name = ['pmdarima==1.8.3']
+    _modules_needed_by_name = ['pmdarima==2.0.4']
     _included_model_classes = None
     _testing_can_skip_failure = False  # ensure tested as if shouldn't fail
     _lag_recipe_allowed = True