mark flaky test as flaky

This commit is contained in:
Ishaan Jaff 2024-08-31 15:02:56 -07:00
parent 2e0ee8c72f
commit 9a3873b9ed

View file

@ -976,6 +976,7 @@ async def test_async_embedding_bedrock():
# CACHING
## Test Azure - completion, embedding
@pytest.mark.asyncio
@pytest.mark.flaky(retries=3, delay=1)
async def test_async_completion_azure_caching():
litellm.set_verbose = True
customHandler_caching = CompletionCustomHandler()