From 36862d0a982ba3a9268a46bd34a5d83294b32b86 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 12 Dec 2024 20:09:14 -0800 Subject: [PATCH] fix testing retry audio test 3 times --- tests/local_testing/test_audio_speech.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/local_testing/test_audio_speech.py b/tests/local_testing/test_audio_speech.py index 9c96a3727f..cb7cad6f2d 100644 --- a/tests/local_testing/test_audio_speech.py +++ b/tests/local_testing/test_audio_speech.py @@ -42,6 +42,7 @@ import litellm ], ) # , @pytest.mark.asyncio +@pytest.mark.flaky(retries=3, delay=1) async def test_audio_speech_litellm(sync_mode, model, api_base, api_key): speech_file_path = Path(__file__).parent / "speech.mp3" @@ -89,6 +90,7 @@ async def test_audio_speech_litellm(sync_mode, model, api_base, api_key): ) @pytest.mark.skip(reason="local only test - we run testing using MockRequests below") @pytest.mark.asyncio +@pytest.mark.flaky(retries=3, delay=1) async def test_audio_speech_litellm_vertex(sync_mode): litellm.set_verbose = True speech_file_path = Path(__file__).parent / "speech_vertex.mp3"