diff --git a/test/integration/agents/vad_agent/test_vad_with_audio.py b/test/integration/agents/vad_agent/test_vad_with_audio.py index 7d10aa3..fd7d4d7 100644 --- a/test/integration/agents/vad_agent/test_vad_with_audio.py +++ b/test/integration/agents/vad_agent/test_vad_with_audio.py @@ -48,6 +48,7 @@ async def test_real_audio(mocker): audio_out_socket = AsyncMock() vad_streamer = Streaming(audio_in_socket, audio_out_socket) + vad_streamer._ready = True for _ in audio_chunks: await vad_streamer.run() diff --git a/test/unit/agents/test_vad_streaming.py b/test/unit/agents/test_vad_streaming.py index 9b38cd0..ab2da0d 100644 --- a/test/unit/agents/test_vad_streaming.py +++ b/test/unit/agents/test_vad_streaming.py @@ -21,7 +21,9 @@ def streaming(audio_in_socket, audio_out_socket): import torch torch.hub.load.return_value = (..., ...) # Mock - return Streaming(audio_in_socket, audio_out_socket) + streaming = Streaming(audio_in_socket, audio_out_socket) + streaming._ready = True + return streaming async def simulate_streaming_with_probabilities(streaming, probabilities: list[float]):