phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

retry flaky tests 3 times

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-28 13:10:47 -07:00
parent b9b07cd932
commit 041764a132
2 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/tests/test_assistants.py
View file

@ -214,6 +214,7 @@ async def test_add_message_litellm(sync_mode, provider):
[True, False], [True, False],
) # ) #
@pytest.mark.asyncio @pytest.mark.asyncio
@pytest.mark.flaky(retries=3, delay=1)
async def test_aarun_thread_litellm(sync_mode, provider, is_streaming): async def test_aarun_thread_litellm(sync_mode, provider, is_streaming):
""" """
- Get Assistants - Get Assistants

1
litellm/tests/test_dynamic_rate_limit_handler.py
View file

@ -120,6 +120,7 @@ async def test_available_tpm(num_projects, dynamic_rate_limit_handler):
@pytest.mark.parametrize("num_projects", [1, 2, 100]) @pytest.mark.parametrize("num_projects", [1, 2, 100])
@pytest.mark.asyncio @pytest.mark.asyncio
@pytest.mark.flaky(retries=3, delay=1)
async def test_available_rpm(num_projects, dynamic_rate_limit_handler): async def test_available_rpm(num_projects, dynamic_rate_limit_handler):
model = "my-fake-model" model = "my-fake-model"
## SET CACHE W/ ACTIVE PROJECTS ## SET CACHE W/ ACTIVE PROJECTS