Add act to mjx viewer.

PiperOrigin-RevId: 629451047 Change-Id: I0baf546d3d91632291e93dec39f4dba7c329d370
google-deepmind · Apr 30, 2024 · 8d1f65c · 8d1f65c
1 parent ec84ddb
commit 8d1f65c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mjx/mujoco/mjx/viewer.py b/mjx/mujoco/mjx/viewer.py
@@ -53,7 +53,7 @@ def _main(argv: Sequence[str]) -> None:
       start = time.time()
 
       # TODO(robotics-simulation): recompile when changing disable flags, etc.
-      dx = dx.replace(ctrl=d.ctrl, xfrc_applied=d.xfrc_applied)
+      dx = dx.replace(ctrl=d.ctrl, act=d.act, xfrc_applied=d.xfrc_applied)
       dx = dx.replace(qpos=d.qpos, qvel=d.qvel, time=d.time)  # handle resets
       mx = mx.tree_replace({
           'opt.gravity': m.opt.gravity,