Bharath2/iLQR

Cannot determine Numba type of <class 'function'>

BrenoKenji01 opened this issue · 1 comments

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?

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.