fix sagemaker test

This commit is contained in:
Ishaan Jaff 2024-08-15 18:23:41 -07:00
parent 953a67ba4c
commit 3407f20b31
2 changed files with 2 additions and 2 deletions

View file

@ -84,7 +84,7 @@ async def test_completion_sagemaker(sync_mode):
@pytest.mark.asyncio()
@pytest.mark.parametrize("sync_mode", [True])
@pytest.mark.parametrize("sync_mode", [False, True])
async def test_completion_sagemaker_stream(sync_mode):
try:
litellm.set_verbose = False