From 52de8db7c90a9de7ac526ff0257ae2715014bc5e Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 4 Jun 2025 21:21:28 -0700 Subject: [PATCH] fix: gemini-2.0-flash-preview-image-generation test --- tests/local_testing/test_amazing_vertex_completion.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py index 892b75ea47c..25b7a9dd9d9 100644 --- a/tests/local_testing/test_amazing_vertex_completion.py +++ b/tests/local_testing/test_amazing_vertex_completion.py @@ -65,6 +65,7 @@ VERTEX_MODELS_TO_NOT_TEST = [ "gemini-2.0-flash-exp", "gemini-2.0-flash-thinking-exp", "gemini-2.0-flash-thinking-exp-01-21", + "gemini-2.0-flash-preview-image-generation", ]