Fix allowdrift and allowmean in non-stepwise AutoARIMA (#828)

Nixtla · Apr 24, 2024 · e46ece9 · e46ece9
1 parent 057cacd
commit e46ece9
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -71,7 +71,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        os: [ubuntu-latest, macos-latest]
+        os: [ubuntu-latest, macos-13]
         python-version: ['3.8', '3.9', '3.10', '3.11']
     steps:
       - name: Clone repo

diff --git a/nbs/src/arima.ipynb b/nbs/src/arima.ipynb
@@ -3014,8 +3014,8 @@
     "            method=method,\n",
     "            xreg=xreg,\n",
     "            offset=offset,\n",
-    "            allowdrift=allowdrift,\n",
-    "            allowmean=allowmean,\n",
+    "            allow_drift=allowdrift,\n",
+    "            allow_mean=allowmean,\n",
     "            period=m,\n",
     "        )\n",
     "        bestfit['lambda'] = blambda\n",

diff --git a/statsforecast/arima.py b/statsforecast/arima.py
@@ -2058,8 +2058,8 @@ def auto_arima_f(
             method=method,
             xreg=xreg,
             offset=offset,
-            allowdrift=allowdrift,
-            allowmean=allowmean,
+            allow_drift=allowdrift,
+            allow_mean=allowmean,
             period=m,
         )
         bestfit["lambda"] = blambda