phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

mark test_completion_sagemaker_prompt_template_non_stream as flaky

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-21 18:14:58 -07:00
parent b987d99486
commit 1ca638973f
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/tests/test_sagemaker.py
View file

@ -301,6 +301,7 @@ async def test_completion_sagemaker_non_stream():
@pytest.mark.asyncio @pytest.mark.asyncio
@pytest.mark.flaky(retries=3, delay=1)
async def test_completion_sagemaker_prompt_template_non_stream(): async def test_completion_sagemaker_prompt_template_non_stream():
mock_response = MagicMock() mock_response = MagicMock()