diff --git a/litellm/tests/test_assistants.py b/litellm/tests/test_assistants.py index 787c38335431..c376eecc2247 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 bc50fc16f50f..f49a760aff7f 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