From 25f60c301b6201b0eeb7bff2787c299f79a0dc40 Mon Sep 17 00:00:00 2001 From: Juan Pizarro Date: Sat, 11 Oct 2025 00:15:42 +0200 Subject: [PATCH] use TeleopEvents.RERECORD_EPISODE in gym_manipulator (#2165) Co-authored-by: Michel Aractingi --- src/lerobot/rl/gym_manipulator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/rl/gym_manipulator.py b/src/lerobot/rl/gym_manipulator.py index ad36f1b36..f9c9d0d7a 100644 --- 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