Skip to content

Commit 552d0eb

Browse files
Merge branch 'user/michel-aractingi/2025_06_30_dataset_v3' into user/michel-aractingi/2025-9-9-incremental_parquet_writing
2 parents b0bc4a3 + 3a7dc18 commit 552d0eb

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

src/lerobot/datasets/lerobot_dataset.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1298,7 +1298,8 @@ def clear_episode_buffer(self, delete_images: bool = True) -> None:
12981298
if self.image_writer is not None:
12991299
self._wait_image_writer()
13001300
episode_index = self.episode_buffer["episode_index"]
1301-
episode_index = episode_index.item() if episode_index.size == 1 else episode_index[0]
1301+
if isinstance(episode_index, np.ndarray):
1302+
episode_index = episode_index.item() if episode_index.size == 1 else episode_index[0]
13021303
for cam_key in self.meta.camera_keys:
13031304
img_dir = self._get_image_file_dir(episode_index, cam_key)
13041305
if img_dir.is_dir():

src/lerobot/datasets/v30/convert_dataset_v21_to_v30.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,8 @@ def convert_data(root: Path, new_root: Path, data_file_size_in_mb: int):
204204
paths_to_cat.append(ep_path)
205205
continue
206206

207-
concat_data_files(paths_to_cat, new_root, chunk_idx, file_idx, image_keys)
207+
if paths_to_cat:
208+
concat_data_files(paths_to_cat, new_root, chunk_idx, file_idx, image_keys)
208209

209210
# Reset for the next file
210211
size_in_mb = ep_size_in_mb

0 commit comments

Comments
 (0)