handle flaky pytests

This commit is contained in:
Ishaan Jaff 2024-08-27 22:43:54 -07:00
parent d8ecf55eca
commit 0f999524c9
5 changed files with 13 additions and 2 deletions

View file

@ -299,6 +299,7 @@ async def _gather_deploy(all_deploys):
@pytest.mark.parametrize(
"ans_rpm", [1, 5]
) # 1 should produce nothing, 10 should select first
@pytest.mark.flaky(retries=3, delay=1)
def test_get_available_endpoints_tpm_rpm_check_async(ans_rpm):
"""
Pass in list of 2 valid models