admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 18:26:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(rollout): clear last_action after return_to_initial

Browse Source
This commit is contained in:
Khalil Meftah
2026-07-02 18:02:36 +02:00
parent 44f76dbbf0
commit 599e2432e5
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/rollout/strategies/dagger.py
+1
View File
@@ -536,6 +536,7 @@ class DAggerStrategy(RolloutStrategy):
engine.pause() engine.pause()
events.phase = DAggerPhase.PAUSED events.phase = DAggerPhase.PAUSED
self._return_to_initial_position(ctx.hardware) self._return_to_initial_position(ctx.hardware)
last_action = None
logger.info( logger.info(
"Episode saved — paused for environment reset. Press SPACE to start next episode." "Episode saved — paused for environment reset. Press SPACE to start next episode."
) )