From fb8fc1a7b6044d906a8e3287d9818504ffd86ba8 Mon Sep 17 00:00:00 2001 From: Maximellerbach Date: Tue, 28 Apr 2026 16:48:30 +0200 Subject: [PATCH] fix crop dataset roi type mismatch Co-authored-by: Copilot --- src/lerobot/rl/crop_dataset_roi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/rl/crop_dataset_roi.py b/src/lerobot/rl/crop_dataset_roi.py index 44614f75f..cc808bcb0 100644 --- a/src/lerobot/rl/crop_dataset_roi.py +++ b/src/lerobot/rl/crop_dataset_roi.py @@ -201,7 +201,7 @@ def convert_lerobot_dataset_to_cropped_lerobot_dataset( # (Here we simply set the shape to be the final resize_size.) for key in crop_params_dict: if key in new_dataset.meta.info.features: - new_dataset.meta.info.features[key]["shape"] = [3] + list(resize_size) + new_dataset.meta.info.features[key]["shape"] = (3, *resize_size) # TODO: Directly modify the mp4 video + meta info features, instead of recreating a dataset prev_episode_index = 0