From e46ece9eaafe80d20a70af58d5fc96edc7737010 Mon Sep 17 00:00:00 2001
From: Manuel Calzolari <2764902+manuel-calzolari@users.noreply.github.com>
Date: Thu, 25 Apr 2024 01:17:29 +0200
Subject: [PATCH] Fix allowdrift and allowmean in non-stepwise AutoARIMA (#828)

---
 .github/workflows/ci.yaml | 2 +-
 nbs/src/arima.ipynb       | 4 ++--
 statsforecast/arima.py    | 4 ++--
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index ee791d5be..c1c332e31 100644
--- 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
index f27809f6e..3e9400940 100644
--- 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
index 82c41c216..8846ebfcc 100644
--- 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