wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add error msg

This commit is contained in:
Simon Alibert 2024-10-24 00:13:53 +02:00
parent 60865e8980
commit 450eae310b
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/common/datasets/lerobot_dataset.py
View File

@ -786,7 +786,9 @@ class LeRobotDataset(torch.utils.data.Dataset):
or shapes is None or shapes is None
or names is None or names is None
): ):
raise ValueError() raise ValueError(
"Dataset info (robot_type, keys, shapes...) must either come from a Robot or explicitly passed upon creation."
)
if len(video_keys) > 0 and not use_videos: if len(video_keys) > 0 and not use_videos:
raise ValueError raise ValueError