Skip to content

Commit 00ef36e

Browse files
committed
Bugfix: Reset recurrent state after episode termination during evaluation in RSL-RL framework
1 parent 79fc49d commit 00ef36e

File tree

3 files changed

+13
-2
lines changed

3 files changed

+13
-2
lines changed

scripts/reinforcement_learning/rsl_rl/play.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -185,7 +185,9 @@ def main(env_cfg: ManagerBasedRLEnvCfg | DirectRLEnvCfg | DirectMARLEnvCfg, agen
185185
# agent stepping
186186
actions = policy(obs)
187187
# env stepping
188-
obs, _, _, _ = env.step(actions)
188+
obs, _, dones, _ = env.step(actions)
189+
# reset recurrent states for episodes that have terminated
190+
policy_nn.reset(dones)
189191
if args_cli.video:
190192
timestep += 1
191193
# Exit the play loop after recording one video

source/isaaclab/config/extension.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
[package]
22

33
# Note: Semantic Versioning is used: https://semver.org/
4-
version = "0.47.2"
4+
version = "0.47.3"
55

66
# Description
77
title = "Isaac Lab framework for Robot Learning"

source/isaaclab/docs/CHANGELOG.rst

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,15 @@
11
Changelog
22
---------
33

4+
0.47.3 (2025-10-26)
5+
~~~~~~~~~~~~~~~~~~~~
6+
7+
Changed
8+
^^^^^^^
9+
10+
* Fixed an issue in recurrent policy evaluation in RSL-RL framework where the recurrent state was not reset after an episode termination.
11+
12+
413
0.47.2 (2025-10-22)
514
~~~~~~~~~~~~~~~~~~~
615

0 commit comments

Comments
 (0)