admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-18 16:57:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix crop dataset roi type mismatch

Browse Source 
Co-authored-by: Copilot <copilot@github.com>
This commit is contained in:
Maximellerbach
2026-04-28 16:48:30 +02:00
parent 16b5010b34
commit fb8fc1a7b6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/rl/crop_dataset_roi.py
+1 -1
View File
@@ -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