[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot]
2025-03-03 06:33:38 +00:00
parent 4e2dc91e59
commit e8126dc3d6
2 changed files with 16 additions and 15 deletions
+1 -3
View File
@@ -707,9 +707,7 @@ class LeRobotDataset(torch.utils.data.Dataset):
item = {}
for vid_key, query_ts in query_timestamps.items():
video_path = self.root / self.meta.get_video_file_path(ep_idx, vid_key)
frames = decode_video_frames_torchcodec(
video_path, query_ts, self.tolerance_s
)
frames = decode_video_frames_torchcodec(video_path, query_ts, self.tolerance_s)
item[vid_key] = frames.squeeze(0)
return item