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

Fix _query_videos return shapes

This commit is contained in:
Simon Alibert 2024-10-23 18:18:28 +02:00
parent b8bdbc1c5b
commit 07570f867f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -487,7 +487,7 @@ class LeRobotDataset(torch.utils.data.Dataset):
frames = decode_video_frames_torchvision( frames = decode_video_frames_torchvision(
video_path, query_ts, self.tolerance_s, self.video_backend video_path, query_ts, self.tolerance_s, self.video_backend
) )
item[vid_key] = frames item[vid_key] = frames.squeeze(0)
return item return item