Skip to content

fix(dataset): reload episodes metadata before batch video encoding #2379

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
wants to merge 6 commits into
base: main
Choose a base branch
from
Draft

fix(dataset): reload episodes metadata before batch video encoding #2379

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
6eed94f
fix(dataset): reload episodes metadata before batch video encoding
daikw Nov 4, 2025
917bfbf
fix(dataset): flush metadata buffer before reloading episodes in batc…
daikw Nov 4, 2025
2bac5ee
fix(dataset): close ParquetWriter before reloading episodes in batch …
daikw Nov 4, 2025
3a26abd
fix(dataset): preserve images for batch encoding and clean up after
daikw Nov 4, 2025
5361df5
fix(dataset): check for video metadata before resume in batch encoding
daikw Nov 4, 2025
d467403
fix(dataset): defer image cleanup and add null check for batch encoding
daikw Nov 4, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions src/lerobot/datasets/lerobot_dataset.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1181,6 +1181,10 @@ def _batch_save_episode_video(self, start_episode: int, end_episode: int | None
f"Batch encoding {self.batch_encoding_size} videos for episodes {start_episode} to {end_episode - 1}"
)

# Reload episodes to ensure we have the latest metadata for all episodes,
# especially when resuming recording with batch encoding enabled
self.meta.episodes = load_episodes(self.root)

chunk_idx = self.meta.episodes[start_episode]["data/chunk_index"]
file_idx = self.meta.episodes[start_episode]["data/file_index"]
episode_df_path = self.root / DEFAULT_EPISODES_PATH.format(chunk_index=chunk_idx, file_index=file_idx)
Expand Down