tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

use TeleopEvents.RERECORD_EPISODE in gym_manipulator (#2165)

Browse Source 
Co-authored-by: Michel Aractingi <michel.aractingi@huggingface.co>
This commit is contained in:
Juan Pizarro
2025-10-11 00:15:42 +02:00
committed by GitHub
parent 0699b46d87
commit 25f60c301b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/lerobot/rl/gym_manipulator.py
View File

@@ -696,7 +696,7 @@ def control_loop(
episode_idx += 1 episode_idx += 1
if dataset is not None: 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}") logging.info(f"Re-recording episode {episode_idx}")
dataset.clear_episode_buffer() dataset.clear_episode_buffer()
episode_idx -= 1 episode_idx -= 1