Make output_vel_repr non-static in api.link

ami-iit · May 23, 2024 · a262dda · a262dda
1 parent a380a22
commit a262dda
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/jaxsim/api/link.py b/src/jaxsim/api/link.py
@@ -203,7 +203,7 @@ def com_in_inertial_frame():
     )
 
 
-@functools.partial(jax.jit, static_argnames=["output_vel_repr"])
+@jax.jit
 def jacobian(
     model: js.model.JaxSimModel,
     data: js.data.JaxSimModelData,
@@ -295,7 +295,7 @@ def jacobian(
     return O_J_WL_I
 
 
-@functools.partial(jax.jit, static_argnames=["output_vel_repr"])
+@jax.jit
 def velocity(
     model: js.model.JaxSimModel,
     data: js.data.JaxSimModelData,