fix: njit with args (#137)

jmoralez · web-flow · commit 4bb140bd3f60 · 2024-11-22T11:27:23.000-06:00
diff --git a/nbs/compat.ipynb b/nbs/compat.ipynb
@@ -84,7 +84,7 @@
     "        return new_dec\n",
     "    \n",
     "    @_doublewrap\n",
-    "    def njit(f):\n",
+    "    def njit(f, *_args, **_kwargs):\n",
     "        @wraps(f)\n",
     "        def wrapper(*args, **kwargs):\n",
     "            warnings.warn(\n",
diff --git a/settings.ini b/settings.ini
@@ -1,7 +1,7 @@
 [DEFAULT]
 repo = utilsforecast
 lib_name = utilsforecast
-version = 0.2.8
+version = 0.2.9
 min_python = 3.8
 license = apache2
 black_formatting = True
diff --git a/utilsforecast/__init__.py b/utilsforecast/__init__.py
@@ -1 +1 @@
-__version__ = "0.2.8"
+__version__ = "0.2.9"
diff --git a/utilsforecast/compat.py b/utilsforecast/compat.py
@@ -61,7 +61,7 @@ def new_dec(*args, **kwargs):
         return new_dec
 
     @_doublewrap
-    def njit(f):
+    def njit(f, *_args, **_kwargs):
         @wraps(f)
         def wrapper(*args, **kwargs):
             warnings.warn(
