decorator not implemented

[gviejo]

if decorated:
convert_to_numpy
run the function
convert back to jax

[gviejo]

def decoratorcasting(func):
	
	def wrapper(array):		
		return func(np.asarray(array))

	return wrapper

@decoratorcasting
@jit(nopython=True)
def test_numba(array):
	return np.sum(array)

[gviejo]

grabs the configuration