Disable AV1 test in fbcode (#463)

pytorch · Jan 22, 2025 · 547dd36 · 547dd36
1 parent 64d1b0d
commit 547dd36
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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)