phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: cleanup tests

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-18 11:01:00 -07:00
parent 3f7252c422
commit ae9f1a0e52
2 changed files with 1 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_amazing_vertex_completion.py
View file

@ -1056,8 +1056,6 @@ def test_gemini_pro_vision_async():
print(resp) print(resp)
asyncio.run(test()) asyncio.run(test())
assert False
except litellm.RateLimitError: except litellm.RateLimitError:
pass pass
except Exception as e: except Exception as e:

2
litellm/tests/test_completion.py
View file

@ -697,7 +697,7 @@ def test_completion_claude_3_base64():
@pytest.mark.parametrize( @pytest.mark.parametrize(
"model", ["gemini/gemini-1.5-flash"] # "claude-3-sonnet-20240229", "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 litellm.set_verbose = True
image_content = [ image_content = [