[bugfix] Fix the output format for VideoClips.subset (#6700)

YosuaMichael · datumbox · web-flow · commit a46c4f0ccdb6 · 2022-10-05T13:54:05.000+01:00
Co-authored-by: Vasilis Vryniotis &lt;datumbox@users.noreply.github.com&gt;
diff --git a/torchvision/datasets/video_utils.py b/torchvision/datasets/video_utils.py
@@ -198,6 +198,7 @@ def subset(self, indices: List[int]) -> "VideoClips":
             _video_max_dimension=self._video_max_dimension,
             _audio_samples=self._audio_samples,
             _audio_channels=self._audio_channels,
+            output_format=self.output_format,
         )
 
     @staticmethod

Original file line number	Diff line number	Diff line change
`@@ -198,6 +198,7 @@ def subset(self, indices: List[int]) -> "VideoClips":`
`198`	`198`	`_video_max_dimension=self._video_max_dimension,`
`199`	`199`	`_audio_samples=self._audio_samples,`
`200`	`200`	`_audio_channels=self._audio_channels,`
	`201`	`+ output_format=self.output_format,`
`201`	`202`	`)`
`202`	`203`
`203`	`204`	`@staticmethod`