From d15ed86e3e7312d874ac758a3565acce0d44263f Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 23 Jan 2025 18:10:31 -0800 Subject: [PATCH] test_chat_completion_ratelimit add retry on test --- tests/test_openai_endpoints.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_openai_endpoints.py b/tests/test_openai_endpoints.py index 6706e4d9f73..77c7c175c9f 100644 --- a/tests/test_openai_endpoints.py +++ b/tests/test_openai_endpoints.py @@ -297,6 +297,7 @@ async def test_chat_completion(): @pytest.mark.asyncio +@pytest.mark.flaky(retries=3, delay=1) async def test_chat_completion_ratelimit(): """ - call model with rpm 1