fix: execstack

chidiwilliams · Jan 4, 2024 · a2ef416 · a2ef416
1 parent c70b594
commit a2ef416
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/mock_sounddevice.py b/tests/mock_sounddevice.py
@@ -6,7 +6,8 @@
 
 import numpy as np
 import sounddevice
-import whisper
+
+from buzz import whisper_audio
 
 mock_query_devices = [
     {
@@ -114,11 +115,11 @@ def start(self):
         self.thread.start()
 
     def target(self):
-        sample_rate = whisper.audio.SAMPLE_RATE
+        sample_rate = whisper_audio.SAMPLE_RATE
         file_path = os.path.join(
             os.path.dirname(__file__), "../testdata/whisper-french.mp3"
         )
-        audio = whisper.load_audio(file_path, sr=sample_rate)
+        audio = whisper_audio.load_audio(file_path, sr=sample_rate)
 
         chunk_duration_secs = 1