From 041764a13240ecf4065e24a67a8efb369eeaaa2e Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 28 Aug 2024 13:10:47 -0700 Subject: [PATCH] retry flaky tests 3 times --- litellm/tests/test_assistants.py | 1 + litellm/tests/test_dynamic_rate_limit_handler.py | 1 + 2 files changed, 2 insertions(+) diff --git a/litellm/tests/test_assistants.py b/litellm/tests/test_assistants.py index 787c383354..c376eecc22 100644 --- a/litellm/tests/test_assistants.py +++ b/litellm/tests/test_assistants.py @@ -214,6 +214,7 @@ async def test_add_message_litellm(sync_mode, provider): [True, False], ) # @pytest.mark.asyncio +@pytest.mark.flaky(retries=3, delay=1) async def test_aarun_thread_litellm(sync_mode, provider, is_streaming): """ - Get Assistants diff --git a/litellm/tests/test_dynamic_rate_limit_handler.py b/litellm/tests/test_dynamic_rate_limit_handler.py index bc50fc16f5..f49a760aff 100644 --- a/litellm/tests/test_dynamic_rate_limit_handler.py +++ b/litellm/tests/test_dynamic_rate_limit_handler.py @@ -120,6 +120,7 @@ async def test_available_tpm(num_projects, dynamic_rate_limit_handler): @pytest.mark.parametrize("num_projects", [1, 2, 100]) @pytest.mark.asyncio +@pytest.mark.flaky(retries=3, delay=1) async def test_available_rpm(num_projects, dynamic_rate_limit_handler): model = "my-fake-model" ## SET CACHE W/ ACTIVE PROJECTS