def wrapper(*args, **kwargs):
spec = inspect.getargspec(f)
received_arg_count = len(args)
received_kwargs = kwargs.copy()
defs = list(spec.defaults)
for arg in spec.args:
if received_arg_count:
received_arg_count -= 1
elif arg not in received_kwargs:
kwargs[arg] = defs.pop(0)
else:
raise Exception("Value not provided for argument " + arg + ".")
args = list(args)
for i, arg in enumerate(args):
if _is_jumpy(arg):
args[i] = arg.array
for k in kwargs:
v = kwargs[k]
if _is_jumpy(v):
kwargs[k] = v.array
out = f(*args, **kwargs)
if _is_nd4j(out):
return array(out)
elif type(out) is list:
for i, v in enumerate(out):
if _is_nd4j(v):
out[i] = array(v)
return out
elif type(out) is tuple:
out = list(out)
for i, v in enumerate(out):
if _is_nd4j(v):
out[i] = array(v)
return tuple(out)
def wrapper(*args, **kwargs):
args = list(args)
for i, arg in enumerate(args):
if _is_jumpy(arg):
args[i] = arg.array
elif type(arg) is list:
for j, a in enumerate(arg):
if _is_jumpy(a):
arg[j] = a.array
elif type(arg) is tuple:
arg = list(arg)
for j, a in enumerate(arg):
if _is_jumpy(a):
arg[j] = a.array
args[i] = tuple(arg)
for k in kwargs:
v = kwargs[k]
if _is_jumpy(v):
kwargs[k] = v.array
out = f(*args, **kwargs)
if _is_nd4j(out):
return array(out)
elif type(out) is list:
for i, v in enumerate(out):
if _is_nd4j(v):
out[i] = array(v)
return out
elif type(out) is tuple:
out = list(out)
for i, v in enumerate(out):
if _is_nd4j(v):
out[i] = array(v)
return tuple(out)
def wrapper(*args, **kwargs):
args = list(args)
for i, arg in enumerate(args):
if _is_jumpy(arg):
args[i] = arg.array
elif type(arg) is list:
for j, a in enumerate(arg):
if _is_jumpy(a):
arg[j] = a.array
elif type(arg) is tuple:
arg = list(arg)
for j, a in enumerate(arg):
if _is_jumpy(a):
arg[j] = a.array
args[i] = tuple(arg)
for k in kwargs:
v = kwargs[k]
if _is_jumpy(v):
kwargs[k] = v.array
out = f(*args, **kwargs)
if _is_nd4j(out):
return array(out)
elif type(out) is list:
for i, v in enumerate(out):
if _is_nd4j(v):
out[i] = array(v)
return out
elif type(out) is tuple:
out = list(out)
for i, v in enumerate(out):
if _is_nd4j(v):
out[i] = array(v)
return tuple(out)