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.
- 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():
|