Cannot determine Numba type of <class 'function'>
BrenoKenji01 opened this issue · 1 comments
BrenoKenji01 commented
Hi,
I am using a python environment named Firedrake (PDE solver) together with another libraries called pygmsh to implement a dynamical system inside the function f(x,u). When I try to run the code, it appears that one of my utility functions has an incompatibility with Numba, even though it doesn't use Numba at all.
Is there any way to make Numba invoked by ilqr ignore my functions?
Bharath2 commented
Try implementing your function like this:
@numba.njit
def f(x, u):
#code
with objmode():
# use firedrake
#code
return x_dot
let me know, if it works.