test: fix test

2024-08-27 07:12:48 -07:00 · 2024-08-27 07:12:48 -07:00 · b0f01e5b95
commit b0f01e5b95
parent ec2357f1d2
1 changed files with 1 additions and 3 deletions
--- 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