Georg commited on
Commit
0aa95e0
·
1 Parent(s): c49a482

Reset UI environment on episode control requests in mujoco_server.py

Browse files

- Added functionality to reset the environment automatically when an episode control action is requested via the UI, allowing users to continue without manual resets.
- This enhancement improves user experience by streamlining the episode management process.

Files changed (1) hide show
  1. mujoco_server.py +4 -0
mujoco_server.py CHANGED
@@ -580,6 +580,10 @@ def _signal_episode_control(action: str):
580
  "message": f"Episode {status_label} requested via UI",
581
  "timestamp": payload["timestamp"],
582
  })
 
 
 
 
583
 
584
 
585
  def _consume_episode_control_flags():
 
580
  "message": f"Episode {status_label} requested via UI",
581
  "timestamp": payload["timestamp"],
582
  })
583
+ # Reset UI environment so the user can continue without manual reset
584
+ with mujoco_lock:
585
+ if env is not None:
586
+ env.reset()
587
 
588
 
589
  def _consume_episode_control_flags():