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: