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): guard empty buffer save

Browse Source
This commit is contained in:
Khalil Meftah
2026-07-02 18:02:59 +02:00
parent 599e2432e5
commit b29ba27977
1 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/rollout/strategies/dagger.py
+14 -10
View File
@@ -556,11 +556,13 @@ class DAggerStrategy(RolloutStrategy):
logger.info("DAgger continuous control loop ended — pausing engine") logger.info("DAgger continuous control loop ended — pausing engine")
engine.pause() engine.pause()
with contextlib.suppress(Exception): with contextlib.suppress(Exception):
with self._episode_lock: buf = dataset.writer.episode_buffer
self._stamp_episode_success(dataset) if buf and any(len(v) > 0 for v in buf.values() if isinstance(v, list)):
dataset.save_episode() with self._episode_lock:
self._needs_push.set() self._stamp_episode_success(dataset)
logger.info("Final in-progress episode saved") dataset.save_episode()
self._needs_push.set()
logger.info("Final in-progress episode saved")
# ------------------------------------------------------------------ # ------------------------------------------------------------------
# Corrections-only mode (record_autonomous=False) # Corrections-only mode (record_autonomous=False)
@@ -708,11 +710,13 @@ class DAggerStrategy(RolloutStrategy):
logger.info("DAgger corrections-only loop ended — pausing engine") logger.info("DAgger corrections-only loop ended — pausing engine")
engine.pause() engine.pause()
with contextlib.suppress(Exception): with contextlib.suppress(Exception):
with self._episode_lock: buf = dataset.writer.episode_buffer
self._stamp_episode_success(dataset) if buf and any(len(v) > 0 for v in buf.values() if isinstance(v, list)):
dataset.save_episode() with self._episode_lock:
self._needs_push.set() self._stamp_episode_success(dataset)
logger.info("Final in-progress episode saved") dataset.save_episode()
self._needs_push.set()
logger.info("Final in-progress episode saved")
# ------------------------------------------------------------------ # ------------------------------------------------------------------
# State-machine transition side-effects # State-machine transition side-effects