scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 86

test(test_completion_cost.py): cleanup test

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-04 16:57:25 -07:00
parent 52a2f5150c
commit 15b226ca4f
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_completion_cost.py
View file

@ -472,7 +472,7 @@ def test_replicate_llama3_cost_tracking():
assert cost == expected_cost
@pytest.mark.parametrize("is_streaming", [True]) # False
@pytest.mark.parametrize("is_streaming", [True, False]) #
def test_groq_response_cost_tracking(is_streaming):
from litellm.utils import (
ModelResponse,
@ -517,4 +517,3 @@ def test_groq_response_cost_tracking(is_streaming):
assert response_cost > 0.0
print(f"response_cost: {response_cost}")
assert False