Fix tests

jordimas · jordimas · commit e05e76c23a2b · 2025-01-12T21:14:02.000+01:00
diff --git a/tests/audio_processing_test.py b/tests/audio_processing_test.py
@@ -24,7 +24,7 @@
 
 from moviepy.audio.AudioClip import AudioArrayClip
 from pyannote.audio import Pipeline
-from pydub import AudioSegment
+from open_dubbing.pydub_audio_segment import AudioSegment
 
 from open_dubbing import audio_processing
 
diff --git a/tests/text_to_speech_test.py b/tests/text_to_speech_test.py
@@ -21,8 +21,7 @@
 
 import pytest
 
-from pydub import AudioSegment
-
+from open_dubbing.pydub_audio_segment import AudioSegment
 from open_dubbing.text_to_speech import TextToSpeech, Voice
 
 
