Files
lerobot/examples/dataset
CarolinePascal 9372f52fff feat(datasets): optionally rewrite per-episode stats when recomputing
Add an opt-in update_episode_stats flag so recomputing stats can also rewrite
the per-episode stats/* columns in the episodes parquet, keeping them consistent
with meta/stats.json. compute_dataset_episode_stats now returns a {episode_index:
stats} mapping so stats can be written back to the right episode and shards merge
by key. Wired into recompute_stats, the lerobot-edit-dataset CLI, and the SLURM
example (per-episode shards + --update-episode-stats).
2026-07-10 23:35:08 +02:00
..