fix: pass constant in autoarima stepwise

Nixtla · Sep 6, 2024 · cf05bd5 · cf05bd5
1 parent 4bd4364
commit cf05bd5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/nbs/src/arima.ipynb b/nbs/src/arima.ipynb
@@ -2618,6 +2618,7 @@
     "        fit = p_myarima(\n",
     "            order=(p, d, q),\n",
     "            seasonal={'order': (P, D, Q), 'period': m},\n",
+    "            constant=constant,\n",
     "        )\n",
     "        results[k] = (p, d, q, P, D, Q, constant, fit['ic'])\n",
     "        k += 1\n",

diff --git a/python/statsforecast/arima.py b/python/statsforecast/arima.py
@@ -1687,6 +1687,7 @@ def try_params(p, d, q, P, D, Q, constant, k, bestfit):
         fit = p_myarima(
             order=(p, d, q),
             seasonal={"order": (P, D, Q), "period": m},
+            constant=constant,
         )
         results[k] = (p, d, q, P, D, Q, constant, fit["ic"])
         k += 1