From ae9f1a0e52cec940cfddaa4cb5465a038c841e70 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 18 Jun 2024 11:01:00 -0700 Subject: [PATCH] test: cleanup tests --- litellm/tests/test_amazing_vertex_completion.py | 2 -- litellm/tests/test_completion.py | 2 +- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index 60ca54abf9..d41c0a916f 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -1056,8 +1056,6 @@ def test_gemini_pro_vision_async(): print(resp) asyncio.run(test()) - - assert False except litellm.RateLimitError: pass except Exception as e: diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 4d69b9ae56..bcbe6753c0 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -697,7 +697,7 @@ def test_completion_claude_3_base64(): @pytest.mark.parametrize( "model", ["gemini/gemini-1.5-flash"] # "claude-3-sonnet-20240229", ) -def test_completion_claude_3_function_plus_image(model): +def test_completion_function_plus_image(model): litellm.set_verbose = True image_content = [