Fix

Softcatala · Nov 29, 2024 · 882337b · 882337b
1 parent 6bcdda4
commit 882337b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Makefile b/Makefile
@@ -23,7 +23,7 @@ update-e2e-tests:
 	$(E2E_COMMAND) e2e-tests/gossos.mp3 --device cpu  --compute_type float32 --output_dir e2e-tests/ref-small-transcribe/
 	$(E2E_COMMAND) e2e-tests/dosparlants.mp3 --temperature_increment_on_fallback None  --device cpu --model medium --compute_type float32 --output_dir e2e-tests/ref-medium-diarization/ --hf_token ${HF_TOKEN}
 	$(E2E_COMMAND) e2e-tests/gossos.mp3 --device cpu --compute_type float32 --max_words_per_line 5 --word_timestamps True --output_dir e2e-tests/ref-small-transcribe-line-max-words/
-	$(E2E_COMMAND) e2e-tests/gossos.mp3 --device cpu --compute_type float32 --vad_filter True --vad_threshold 0.5 --vad_min_speech_duration_ms 2000 --vad_max_speech_duration_s 50000 --output_dir e2e-tests/ref-small-transcribe-vad/
+	$(E2E_COMMAND) e2e-tests/gossos.mp3 --device cpu --compute_type float32 --vad_filter True --vad_onset 0.5 --vad_min_speech_duration_ms 2000 --vad_max_speech_duration_s 50000 --output_dir e2e-tests/ref-small-transcribe-vad/
 
 publish-release:
 	rm dist/ -r -f

diff --git a/e2e-tests/testcmd.py b/e2e-tests/testcmd.py
@@ -164,8 +164,8 @@ def test_options_vad(self):
         with tempfile.TemporaryDirectory() as directory:
             _file = "gossos"
             cmd = (
-                f"cd {directory} && whisper-ctranslate2 {path}/{_file}.mp3 --device cpu --model medium --compute_type float32 --vad_filter True --vad_threshold 0.5"
-                " --vad_min_speech_duration_ms 2000 --vad_max_speech_duration_s 50000  --output_dir {directory}"
+                f"cd {directory} && whisper-ctranslate2 {path}/{_file}.mp3 --device cpu --model medium --compute_type float32 --vad_filter True --vad_onset 0.5"
+                " --vad_min_speech_duration_ms 2000 --vad_max_speech_duration_s 50000 --output_dir {directory}"
             )
             os.system(cmd)
             self._check_ref_small(