use TeleopEvents.RERECORD_EPISODE in gym_manipulator (#2165)

Co-authored-by: Michel Aractingi <michel.aractingi@huggingface.co>
2025-10-11 00:15:42 +02:00
parent 0699b46d87
commit 25f60c301b
1 changed files with 1 additions and 1 deletions
--- a/src/lerobot/rl/gym_manipulator.py
+++ b/src/lerobot/rl/gym_manipulator.py
@@ -696,7 +696,7 @@ def control_loop(
            episode_idx += 1

            if dataset is not None:
-                if transition[TransitionKey.INFO].get("rerecord_episode", False):
+                if transition[TransitionKey.INFO].get(TeleopEvents.RERECORD_EPISODE, False):
                    logging.info(f"Re-recording episode {episode_idx}")
                    dataset.clear_episode_buffer()
                    episode_idx -= 1