mark test_transcription as flaky

2024-12-02 23:34:28 -08:00 · 2024-12-02 23:34:28 -08:00 · 61b590f00c
commit 61b590f00c
parent 6e6cd05f69
1 changed files with 1 additions and 0 deletions
--- a/tests/local_testing/test_whisper.py
+++ b/tests/local_testing/test_whisper.py
@ -56,6 +56,7 @@ from litellm import Router
    [("json", None), ("vtt", None), ("verbose_json", ["word"])],
 )
@pytest.mark.asyncio
+@pytest.mark.flaky(retries=3, delay=1)
 async def test_transcription(
    model, api_key, api_base, response_format, sync_mode, timestamp_granularities
 ):