diff --git a/mjx/mujoco/mjx/_src/smooth.py b/mjx/mujoco/mjx/_src/smooth.py index fba3d3ce..bef176d6 100644 --- a/mjx/mujoco/mjx/_src/smooth.py +++ b/mjx/mujoco/mjx/_src/smooth.py @@ -1191,6 +1191,9 @@ def tendon_armature(m: Model, d: Data) -> Data: if not isinstance(m._impl, ModelJAX) or not isinstance(d._impl, DataJAX): raise ValueError('tendon_armature requires JAX backend implementation.') + if not m.ntendon: + return d + if not support.is_sparse(m): return d.tree_replace({ '_impl.qM': (