ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki 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
Show Stats 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.flaky(retries=3, delay=1)
async def test_completion_sagemaker_prompt_template_non_stream():
mock_response = MagicMock()