From 3675745fab7b6e22c481b246eff777f829ef3e2a Mon Sep 17 00:00:00 2001
From: Monesh Ram <31161039+WhoisMonesh@users.noreply.github.com>
Date: Sat, 21 Feb 2026 18:02:59 +0530
Subject: [PATCH] Update
 tests/litellm/llms/vertex_ai/gemini/test_transformation.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
---
 tests/litellm/llms/vertex_ai/gemini/test_transformation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/litellm/llms/vertex_ai/gemini/test_transformation.py b/tests/litellm/llms/vertex_ai/gemini/test_transformation.py
index f19feca0861..b05dd64df1f 100644
--- a/tests/litellm/llms/vertex_ai/gemini/test_transformation.py
+++ b/tests/litellm/llms/vertex_ai/gemini/test_transformation.py
@@ -1,4 +1,4 @@
-from litellm.llms.vertex_ai.gemini.transformation import VertexGeminiConfig
+from litellm.llms.vertex_ai.gemini.vertex_and_google_ai_studio_gemini import VertexGeminiConfig
 
 def test_vertex_ai_audio_supported_params():
     config = VertexGeminiConfig()