From 827ce52d802d9ef7f22c4b7d50a5e7cf9b5ceee3 Mon Sep 17 00:00:00 2001 From: yuneng-jiang Date: Thu, 22 Jan 2026 15:21:44 -0800 Subject: [PATCH] Adding retries to flaky tests --- tests/test_keys.py | 2 +- tests/test_organizations.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_keys.py b/tests/test_keys.py index 3843a79eb4a..5b269d08894 100644 --- a/tests/test_keys.py +++ b/tests/test_keys.py @@ -583,7 +583,7 @@ async def test_aaaaakey_info_spend_values_streaming(): rounded_response_cost == rounded_key_info_spend ), f"Expected={rounded_response_cost}, Got={rounded_key_info_spend}" - +@pytest.mark.flaky(retries=3, delay=1) @pytest.mark.asyncio async def test_key_info_spend_values_image_generation(): """ diff --git a/tests/test_organizations.py b/tests/test_organizations.py index 46281b4789d..ddb48508be3 100644 --- a/tests/test_organizations.py +++ b/tests/test_organizations.py @@ -188,7 +188,7 @@ async def list_organization(session, i): return response_json - +@pytest.mark.flaky(retries=5, delay=1) @pytest.mark.asyncio async def test_organization_new(): """