diff --git a/test/decoders/test_video_decoder.py b/test/decoders/test_video_decoder.py index 2fcd324d..3cbacd9f 100644 --- a/test/decoders/test_video_decoder.py +++ b/test/decoders/test_video_decoder.py @@ -418,6 +418,9 @@ def test_get_frames_at_fails(self, device): @pytest.mark.parametrize("device", cpu_and_cuda()) def test_get_frame_at_av1(self, device): + if in_fbcode() and device == "cuda": + return + decoder = VideoDecoder(AV1_VIDEO.path, device=device) ref_frame10 = AV1_VIDEO.get_frame_data_by_index(10) ref_frame_info10 = AV1_VIDEO.get_frame_info(10)