diff --git a/tests/collections/asr/test_diar_neural_inference.py b/tests/collections/asr/test_diar_neural_inference.py index 076eac129293..99dc1b9586a1 100644 --- 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])