Skip to content

Commit d51b7a5

Browse files
Fix type hint for tts_engine parameter and update mock dependencies in audio tests
1 parent d06f8c7 commit d51b7a5

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

src/sdialog/audio/dialog.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -428,7 +428,7 @@ def persona_to_voice(
428428
persona_to_voice_desc: Union[str, callable] = None,
429429
voices: dict[Role, Union[Voice, tuple[str, str]]] = None,
430430
keep_duplicate: bool = False,
431-
tts_engine: BaseTTS | BaseVoiceCloneTTS = None,
431+
tts_engine: Union[BaseTTS, BaseVoiceCloneTTS] = None,
432432
seed: int = None
433433
) -> None:
434434
"""

tests/test_audio.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -704,14 +704,17 @@ def mock_dependencies():
704704
with patch('sdialog.audio.pipeline.Qwen3TTS') as mock_tts, \
705705
patch('sdialog.audio.pipeline.Qwen3TTSVoiceClone') as mock_tts, \
706706
patch('sdialog.audio.pipeline.HuggingfaceVoiceDatabase') as mock_db, \
707-
patch('sdialog.audio.pipeline.scaper', create=True) as mock_scaper, \
707+
patch('sdialog.audio.pipeline.dscaper') as mock_dscaper, \
708+
patch('sdialog.audio.pipeline.load_dataset', return_value=[]) as mock_load_dataset, \
709+
patch('sdialog.audio.pipeline.default_dscaper_datasets', return_value=[]) as mock_default_datasets, \
708710
patch('sdialog.audio.pipeline.generate_utterances_audios') as mock_gen_utt, \
709711
patch('sdialog.audio.dialog.AudioDialog.save_utterances_audios') as mock_save_utt, \
710712
patch('sdialog.audio.pipeline.librosa', create=True) as mock_librosa, \
711713
patch('sdialog.audio.pipeline.sf', create=True) as mock_sf, \
712714
patch('sdialog.audio.pipeline.generate_audio_room_accoustic') as mock_gen_room:
713715
yield {
714-
"tts": mock_tts, "db": mock_db, "scaper": mock_scaper,
716+
"tts": mock_tts, "db": mock_db, "dscaper": mock_dscaper,
717+
"load_dataset": mock_load_dataset, "default_dscaper_datasets": mock_default_datasets,
715718
"gen_utt": mock_gen_utt, "save_utt": mock_save_utt,
716719
"librosa": mock_librosa, "sf": mock_sf, "gen_room": mock_gen_room,
717720
"ir_db": MagicMock()

0 commit comments

Comments
 (0)