🔨 refactor whisper_live according to flake8
This commit is contained in:
+1
-2
@@ -1,7 +1,6 @@
|
||||
import unittest
|
||||
import numpy as np
|
||||
import torch
|
||||
import scipy.io as sio
|
||||
from whisper_live.tensorrt_utils import load_audio
|
||||
from whisper_live.vad import VoiceActivityDetection
|
||||
|
||||
@@ -25,4 +24,4 @@ class TestVoiceActivityDetection(unittest.TestCase):
|
||||
def test_vad_speech_detection(self):
|
||||
audio_tensor = torch.from_numpy(load_audio("assets/jfk.flac"))
|
||||
speech_prob = self.vad(audio_tensor, self.sample_rate).item()
|
||||
self.assertGreater(speech_prob, 0.5, "VAD failed to identify speech segment.")
|
||||
self.assertGreater(speech_prob, 0.5, "VAD failed to identify speech segment.")
|
||||
|
||||
Reference in New Issue
Block a user