mypy

NicolasHug · NicolasHug · commit f69760e35adf · 2025-03-05T14:34:42.000Z
diff --git a/src/torchcodec/decoders/_video_decoder.py b/src/torchcodec/decoders/_video_decoder.py
@@ -112,6 +112,8 @@ def __init__(
             decoder=self._decoder, stream_index=stream_index, media_type="video"
         )
 
+        assert isinstance(self.metadata, core.VideoStreamMetadata)  # mypy
+
         if self.metadata.num_frames is None:
             raise ValueError(
                 "The number of frames is unknown. " + ERROR_REPORTING_INSTRUCTIONS

Original file line number	Diff line number	Diff line change
`@@ -112,6 +112,8 @@ def __init__(`
`112`	`112`	`decoder=self._decoder, stream_index=stream_index, media_type="video"`
`113`	`113`	`)`
`114`	`114`
	`115`	`+ assert isinstance(self.metadata, core.VideoStreamMetadata) # mypy`
	`116`	`+`
`115`	`117`	`if self.metadata.num_frames is None:`
`116`	`118`	`raise ValueError(`
`117`	`119`	`"The number of frames is unknown. " + ERROR_REPORTING_INSTRUCTIONS`