From 15b226ca4f7e31cea01bea180b0925d550efb8e6 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 4 Jun 2024 16:57:25 -0700 Subject: [PATCH] test(test_completion_cost.py): cleanup test --- litellm/tests/test_completion_cost.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/litellm/tests/test_completion_cost.py b/litellm/tests/test_completion_cost.py index 2b56934850d..7820e2af3d2 100644 --- a/litellm/tests/test_completion_cost.py +++ b/litellm/tests/test_completion_cost.py @@ -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