Skip to content

Commit 7291466

Browse files
committed
Fixed mypy issue in VideoClips
1 parent 15c166a commit 7291466

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

torchvision/datasets/video_utils.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ def __init__(
136136

137137
def _compute_frame_pts(self) -> None:
138138
self.video_pts = []
139-
self.video_fps = []
139+
self.video_fps: List[int] = []
140140

141141
# strategy: use a DataLoader to parallelize read_video_timestamps
142142
# so need to create a dummy dataset first
@@ -160,6 +160,8 @@ def _compute_frame_pts(self) -> None:
160160
self.video_pts.extend(clips)
161161
self.video_fps.extend(fps)
162162

163+
print("self.video_fps:", [type(i) for i in self.video_fps])
164+
163165
def _init_from_metadata(self, metadata: Dict[str, Any]) -> None:
164166
self.video_paths = metadata["video_paths"]
165167
assert len(self.video_paths) == len(metadata["video_pts"])

0 commit comments

Comments
 (0)