Fix identity decorator for when Numba unavailable

moble · Aug 10, 2023 · 17ed6a2 · 17ed6a2
1 parent 7113a90
commit 17ed6a2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/quaternion/numba_wrapper.py b/src/quaternion/numba_wrapper.py
@@ -21,9 +21,9 @@
     #     "may wish to install numba." + \
     #     "\n" + "!" * 53 + "\n"
     # warnings.warn(warning_text)
-    def _identity_decorator_outer(*args, **kwargs):
-        def _identity_decorator_inner(fn):
-            return fn
+    def _identity_decorator_outer(fn):
+        def _identity_decorator_inner(*args, **kwargs):
+           return fn(*args, **kwargs)
         return _identity_decorator_inner
     njit = _identity_decorator_outer
     jit = _identity_decorator_outer