From 2da89a0c8e9b6ceb1ca23eb0b14791c16bab75ec Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 20 May 2024 10:51:25 -0700 Subject: [PATCH] fix vertex test --- litellm/tests/test_image_generation.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/litellm/tests/test_image_generation.py b/litellm/tests/test_image_generation.py index 886953c1a..6acb28e5b 100644 --- a/litellm/tests/test_image_generation.py +++ b/litellm/tests/test_image_generation.py @@ -173,6 +173,9 @@ async def test_aimage_generation_bedrock_with_optional_params(): @pytest.mark.asyncio async def test_aimage_generation_vertex_ai(): + from test_amazing_vertex_completion import load_vertex_ai_credentials + + load_vertex_ai_credentials() try: response = await litellm.aimage_generation( prompt="An olympic size swimming pool",