From 1d0eb922bdf79c77bf4c97098873d32cffea9f82 Mon Sep 17 00:00:00 2001 From: fracapuano Date: Fri, 7 Nov 2025 21:06:51 +0000 Subject: [PATCH] fix: episode index is asserted 0-based in lerobot dataset --- examples/behavior_1k/convert_to_lerobot_v3.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/examples/behavior_1k/convert_to_lerobot_v3.py b/examples/behavior_1k/convert_to_lerobot_v3.py index b76f24a6b..c09dba3d1 100755 --- a/examples/behavior_1k/convert_to_lerobot_v3.py +++ b/examples/behavior_1k/convert_to_lerobot_v3.py @@ -202,6 +202,12 @@ def convert_data(root: Path, new_root: Path, data_file_size_in_mb: int, task_ind size_in_mb += ep_size_in_mb num_frames += ep_num_frames episodes_metadata.append(ep_metadata) + + # write 0-based episode index instead of custom episode index (otherwise breaks compatibility with LeRobotDataset) + tmp_df = pd.read_parquet(ep_path) + tmp_df["episode_index"] = ep_idx + tmp_df.to_parquet(ep_path) + ep_idx += 1 if size_in_mb < data_file_size_in_mb: