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(): """