jhj0517 commited on
Commit
60aa217
·
1 Parent(s): 88d2794

Skip diarization pipeline for CPU environment

Browse files
Files changed (1) hide show
  1. tests/test_diarization.py +4 -0
tests/test_diarization.py CHANGED
@@ -9,6 +9,10 @@ import pytest
9
  import os
10
 
11
 
 
 
 
 
12
  @pytest.mark.parametrize(
13
  "whisper_type,vad_filter,bgm_separation,diarization",
14
  [
 
9
  import os
10
 
11
 
12
+ @pytest.mark.skipif(
13
+ not is_cuda_available(),
14
+ reason="Skipping because the test only works on GPU"
15
+ )
16
  @pytest.mark.parametrize(
17
  "whisper_type,vad_filter,bgm_separation,diarization",
18
  [