See if tests fail without cuda

Signed-off-by: SeanNaren <[email protected]>
NVIDIA · Feb 22, 2023 · d27ce29 · d27ce29
1 parent 8bfe05f
commit d27ce29
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/tests/collections/asr/test_diar_neural_inference.py b/tests/collections/asr/test_diar_neural_inference.py
@@ -23,14 +23,7 @@
 class TestNeuralDiarizerInference:
     @pytest.mark.unit
     @pytest.mark.parametrize(
-        "device",
-        [
-            torch.device("cpu"),
-            pytest.param(
-                torch.device("cuda"),
-                marks=pytest.mark.skipif(not torch.cuda.is_available(), reason='CUDA required for test.',),
-            ),
-        ],
+        "device", [torch.device("cpu"),],
     )
     @pytest.mark.parametrize("num_speakers", [None, 1])
     @pytest.mark.parametrize("max_num_speakers", [4])