Commit 290b447a authored by Amir MOHAMMADI's avatar Amir MOHAMMADI

Merge branch 'issue-64' into 'master'

[] implement array_jit decorator correctly

Closes #64

See merge request !104
parents a7c665ec aedd0cf4
Pipeline #52343 passed with stages
in 14 minutes and 23 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)
def new_func(*args, **kwargs):
args, kwargs = _lists_to_arrays(*args, **kwargs)
return jit_func(*args, **kwargs)
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment