fix(compat): handle zero offset in shift_array (#480)

Author: jmoralez

mlforecast/__init__.py

@@ -1,3 +1,3 @@
-__version__ = "1.0.1"
+__version__ = "1.0.2"
 __all__ = ['MLForecast']
 from mlforecast.forecast import MLForecast

mlforecast/compat.py

@@ -41,6 +41,8 @@ def __init__(self, *args, **kwargs):  # noqa: ARG002
     def shift_array(x, offset):
         if offset >= x.size or offset < 0:
             return np.full_like(x, np.nan)
+        if offset == 0:
+            return x.copy()
         out = np.empty_like(x)
         out[:offset] = np.nan
         out[offset:] = x[:-offset]

nbs/compat.ipynb

@@ -55,6 +55,8 @@
     "    def shift_array(x, offset):\n",
     "        if offset >= x.size or offset < 0:\n",
     "            return np.full_like(x, np.nan)\n",
+    "        if offset == 0:\n",
+    "            return x.copy()\n",
     "        out = np.empty_like(x)\n",
     "        out[:offset] = np.nan\n",
     "        out[offset:] = x[:-offset]\n",

settings.ini

@@ -8,7 +8,7 @@ author = José Morales
 author_email = [email protected]
 copyright = Nixtla
 branch = main
-version = 1.0.1
+version = 1.0.2
 min_python = 3.9
 audience = Developers
 language = English
@@ -23,7 +23,7 @@ aws_requirements = fsspec[s3]
 gcp_requirements = fsspec[gcs]
 azure_requirements = fsspec[adl]
 polars_requirements = polars[numpy]
-dev_requirements = black>=24 datasetsforecast>=1 gitpython holidays<0.21 lightgbm matplotlib mlflow>=2.10.0 mypy nbdev<2.3.26 numpy>=2 pandas>=2.2.2 pre-commit polars[numpy] pyarrow ruff setuptools statsmodels xgboost
+dev_requirements = black>=24 datasetsforecast>=1 gitpython holidays<0.21 lightgbm<4.6 matplotlib mlflow>=2.10.0 mypy nbdev<2.3.26 numpy>=2 pandas>=2.2.2 pre-commit polars[numpy] pyarrow ruff setuptools statsmodels xgboost
 nbs_path = nbs
 doc_path = _docs
 recursive = True