Avoid unnecessary jax.vmap in rbda.forward_kinematics

ami-iit · Nov 20, 2024 · a4716de · a4716de
1 parent f3b039e
commit a4716de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/jaxsim/rbda/forward_kinematics.py b/src/jaxsim/rbda/forward_kinematics.py
@@ -85,7 +85,7 @@ def propagate_kinematics(
         else [(W_X_i,), None]
     )
 
-    return jax.vmap(Adjoint.to_transform)(W_X_i)
+    return Adjoint.to_transform(W_X_i)
 
 
 def forward_kinematics(