diff --git a/tests/spend_tracking_tests/test_spend_accuracy_tests.py b/tests/spend_tracking_tests/test_spend_accuracy_tests.py index be071f2f0f8..3fc9d12a918 100644 --- a/tests/spend_tracking_tests/test_spend_accuracy_tests.py +++ b/tests/spend_tracking_tests/test_spend_accuracy_tests.py @@ -206,6 +206,7 @@ async def fail_with_diagnostics(session, stage: str, expected: float, observed: ) +@pytest.mark.flaky(reruns=2, reruns_delay=5) @pytest.mark.asyncio async def test_basic_spend_accuracy(): """ diff --git a/tests/store_model_in_db_tests/test_callbacks_in_db.py b/tests/store_model_in_db_tests/test_callbacks_in_db.py index 4a851251a3e..f6bd4196af8 100644 --- a/tests/store_model_in_db_tests/test_callbacks_in_db.py +++ b/tests/store_model_in_db_tests/test_callbacks_in_db.py @@ -71,6 +71,7 @@ async def make_chat_completions_request() -> ChatCompletion: return response +@pytest.mark.flaky(reruns=2, reruns_delay=5) @pytest.mark.asyncio async def test_e2e_langfuse_callbacks_in_db(): diff --git a/tests/test_fallbacks.py b/tests/test_fallbacks.py index bc9aa4c64c8..e2792ac40c4 100644 --- a/tests/test_fallbacks.py +++ b/tests/test_fallbacks.py @@ -132,6 +132,7 @@ async def test_chat_completion_client_fallbacks(has_access): pytest.fail("Expected this to work: {}".format(str(e))) +@pytest.mark.flaky(reruns=2, reruns_delay=5) @pytest.mark.asyncio async def test_chat_completion_with_retries(): """ @@ -156,6 +157,7 @@ async def test_chat_completion_with_retries(): assert headers["x-litellm-max-retries"] == "50" +@pytest.mark.flaky(reruns=2, reruns_delay=5) @pytest.mark.asyncio async def test_chat_completion_with_fallbacks(): """