add warning

sintel-dev · Sep 30, 2023 · 6796029 · 6796029
1 parent fd71382
commit 6796029
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/mlstars/adapters/statsmodels.py b/mlstars/adapters/statsmodels.py
@@ -4,11 +4,12 @@
 from statsmodels.tsa.arima import model
 
 _ARIMA_MODEL_DEPRECATION_WARNING = (
-    "mstatsmodels.tsa.arima_model.Arima is deprecated "
+    "statsmodels.tsa.arima_model.Arima is deprecated "
     "and will be removed in a future version. Please use "
     "statsmodels.tsa.arima.model.ARIMA instead."
 )
 
+
 class ARIMA(object):
     """A Wrapper for the statsmodels.tsa.arima.model.ARIMA class."""
 

diff --git a/mlstars/primitives/statsmodels.tsa.arima_model.Arima.json b/mlstars/primitives/statsmodels.tsa.arima_model.Arima.json
@@ -3,7 +3,7 @@
     "contributors": [
         "Alexander Geiger <[email protected]>"
     ],
-    "description": "ARIMA Model",
+    "description": "ARIMA Model; mstatsmodels.tsa.arima_model.Arima is deprecated and will be removed in a future version. Please use statsmodels.tsa.arima.model.ARIMA instead.",
     "classifiers": {
         "type": "estimator",
         "subtype": "regressor"