From cf2754c4fc951e692bf19cb2843cad60197e824f Mon Sep 17 00:00:00 2001
From: Andrew Scott <84710236+andrewscottm@users.noreply.github.com>
Date: Mon, 4 Nov 2024 09:17:46 +0000
Subject: [PATCH] fix: auto arima xreg rank deficient test (#931)

---
 nbs/src/arima.ipynb           | 2 +-
 python/statsforecast/arima.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/nbs/src/arima.ipynb b/nbs/src/arima.ipynb
index 5984481ca..658fc71fa 100644
--- a/nbs/src/arima.ipynb
+++ b/nbs/src/arima.ipynb
@@ -2385,7 +2385,7 @@
     "        else:\n",
     "            if constant_columns.any():\n",
     "                xregg = xregg[:, ~constant_columns]\n",
-    "            X = np.hstack([np.arange(1, xregg.shape[0]+1).reshape(-1, 1), xregg])\n",
+    "            X = np.hstack([np.ones([xregg.shape[0], 1]), xregg])\n",
     "            X = X[~np.isnan(X).any(1)]\n",
     "            _, sv, _ = np.linalg.svd(X)\n",
     "            if sv.min() / sv.sum() < np.finfo(np.float64).eps:\n",
diff --git a/python/statsforecast/arima.py b/python/statsforecast/arima.py
index 09d4b678e..2e2b67e6a 100644
--- a/python/statsforecast/arima.py
+++ b/python/statsforecast/arima.py
@@ -1446,7 +1446,7 @@ def auto_arima_f(
         else:
             if constant_columns.any():
                 xregg = xregg[:, ~constant_columns]
-            X = np.hstack([np.arange(1, xregg.shape[0] + 1).reshape(-1, 1), xregg])
+            X = np.hstack([np.ones([xregg.shape[0], 1]), xregg])
             X = X[~np.isnan(X).any(1)]
             _, sv, _ = np.linalg.svd(X)
             if sv.min() / sv.sum() < np.finfo(np.float64).eps: