From b0f01e5b9537da4a179cd34d1e471479104bf2a3 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Tue, 27 Aug 2024 07:12:48 -0700
Subject: [PATCH] test: fix test

---
 litellm/tests/test_completion.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index d5179da7a9..b39fbc3b0a 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -917,9 +917,7 @@ def test_completion_claude_3_base64():
             pytest.fail(f"An exception occurred - {str(e)}")
 
 
-@pytest.mark.parametrize(
-    "model", ["gemini/gemini-1.5-flash"]  # "claude-3-sonnet-20240229",
-)
+@pytest.mark.parametrize("model", ["claude-3-sonnet-20240229"])
 def test_completion_function_plus_image(model):
     litellm.set_verbose = True