Commit aedd0cf4 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI

[_library.py] implement array_jit decorator correctly

Fixes #64
parent a7c665ec
Pipeline #52314 passed with stage
in 11 minutes and 15 seconds
......@@ -10,6 +10,7 @@ import numpy
import numpy.linalg
from numba import jit, objmode
import logging
from functools import wraps
logger = logging.getLogger(__name__)
......@@ -26,6 +27,7 @@ def _lists_to_arrays(*args, **kwargs):
def array_jit(func):
jit_func = jit(func, nopython=True)
@wraps(jit_func)
def new_func(*args, **kwargs):
args, kwargs = _lists_to_arrays(*args, **kwargs)
return jit_func(*args, **kwargs)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment