Add device param

pytorch · Jan 10, 2025 · bbad535 · bbad535
1 parent 8b4dd51
commit bbad535
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/decoders/test_video_decoder.py b/test/decoders/test_video_decoder.py
@@ -427,12 +427,12 @@ def test_get_frame_played_at(self, device):
         assert isinstance(decoder.get_frame_played_at(6.02).duration_seconds, float)
 
     @pytest.mark.parametrize("device", cpu_and_cuda())
-    def test_get_frame_played_at_h265(self):
+    def test_get_frame_played_at_h265(self, device):
         # Non-regression test for https://github.com/pytorch/torchcodec/issues/179
         # We don't parametrize with CUDA because the current GPUs on CI do not
         # support x265:
         # https://github.com/pytorch/torchcodec/pull/350#issuecomment-2465011730
-        decoder = VideoDecoder(H265_VIDEO.path)
+        decoder = VideoDecoder(H265_VIDEO.path, device=device)
         ref_frame6 = H265_VIDEO.get_frame_data_by_index(5)
         assert_frames_equal(ref_frame6, decoder.get_frame_played_at(0.5).data)