scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 89

test: mark flaky tests

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-30 07:27:51 -07:00
parent a9ca183021
commit 94db4ec830
2 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_alangfuse.py
View file

@ -255,6 +255,7 @@ audio_file = open(file_path, "rb")
@pytest.mark.asyncio
@pytest.mark.flaky(retries=3, delay=1)
async def test_langfuse_logging_audio_transcriptions(langfuse_client):
"""
Test that creates a trace with masked input and output
@ -289,6 +290,7 @@ async def test_langfuse_logging_audio_transcriptions(langfuse_client):
@pytest.mark.asyncio
@pytest.mark.flaky(retries=3, delay=1)
async def test_langfuse_masked_input_output(langfuse_client):
"""
Test that creates a trace with masked input and output

1
litellm/tests/test_lowest_latency_routing.py
View file

@ -348,6 +348,7 @@ def test_get_available_endpoints_tpm_rpm_check_async(ans_rpm):
@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(ans_rpm):
"""
Pass in list of 2 valid models