Update utils.py

Bharath2 · May 15, 2024 · e7672e0 · e7672e0
1 parent ea3698b
commit e7672e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ilqr/utils.py b/ilqr/utils.py
@@ -100,7 +100,7 @@ def sympy_to_numba(f, args, redu = True):
             if n == 1: f = f.T
             f = sp.Array(f)[0, :]
             f = njit(sp.lambdify(args, f, modules = modules))
-            f_new = lambda *args: np.array(f(*args))
+            f_new = lambda *args: np.asarray(f(*args))
             return njit(f_new)
 
     f = sp.lambdify(args, f, modules = modules)