More test disabling

pytorch · May 30, 2023 · 4bf493c · 4bf493c
1 parent 711bd18
commit 4bf493c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/torchaudio_unittest/prototype/conformer_wav2vec2_test.py b/test/torchaudio_unittest/prototype/conformer_wav2vec2_test.py
@@ -52,6 +52,8 @@ def test_pretrain_cpu_smoke_test(self, model, dtype):
         [torch.float32, torch.float64],
     )
     @skipIfNoCuda
+    # Disabled in CI: https://github.com/pytorch/audio/issues/3376
+    @disabledInCI
     def test_pretrain_cuda_smoke_test(self, model, dtype):
         model = model()
         self._smoke_test(model, torch.device("cuda"), dtype)