From e0f1f18339510222ed2a8515d59b6013778ee72f Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 12 Nov 2024 16:46:21 -0800 Subject: [PATCH] fix img gen basic test --- tests/image_gen_tests/test_image_generation.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/image_gen_tests/test_image_generation.py b/tests/image_gen_tests/test_image_generation.py index 14493c3f0..a9e5fe692 100644 --- a/tests/image_gen_tests/test_image_generation.py +++ b/tests/image_gen_tests/test_image_generation.py @@ -104,7 +104,9 @@ def load_vertex_ai_credentials(): class TestVertexImageGeneration(BaseImageGenTest): def get_base_image_generation_call_args(self) -> dict: - # load_vertex_ai_credentials() + # comment this when running locally + load_vertex_ai_credentials() + litellm.in_memory_llm_clients_cache = {} return { "model": "vertex_ai/imagegeneration@006",