From bbad5359b0ce788210c3f57859a78ace03133998 Mon Sep 17 00:00:00 2001 From: Scott Schneider Date: Fri, 10 Jan 2025 08:45:06 -0800 Subject: [PATCH] Add device param --- test/decoders/test_video_decoder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/decoders/test_video_decoder.py b/test/decoders/test_video_decoder.py index afa61737..8b3fca3e 100644 --- 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)