admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-25 02:36: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
+4
View File
@@ -556,6 +556,8 @@ 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):
buf = dataset.writer.episode_buffer
if buf and any(len(v) > 0 for v in buf.values() if isinstance(v, list)):
with self._episode_lock: with self._episode_lock:
self._stamp_episode_success(dataset) self._stamp_episode_success(dataset)
dataset.save_episode() dataset.save_episode()
@@ -708,6 +710,8 @@ 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):
buf = dataset.writer.episode_buffer
if buf and any(len(v) > 0 for v in buf.values() if isinstance(v, list)):
with self._episode_lock: with self._episode_lock:
self._stamp_episode_success(dataset) self._stamp_episode_success(dataset)
dataset.save_episode() dataset.save_episode()