From 77572964c04fa2d64e3a052f07a1ee6aa973eb7d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 5 Apr 2025 13:13:25 -0700 Subject: [PATCH] test_completion_cost_databricks --- tests/local_testing/test_completion_cost.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/local_testing/test_completion_cost.py b/tests/local_testing/test_completion_cost.py index afa52e67e08..c15922e212d 100644 --- a/tests/local_testing/test_completion_cost.py +++ b/tests/local_testing/test_completion_cost.py @@ -1266,7 +1266,7 @@ def test_completion_cost_prompt_caching(model, custom_llm_provider): @pytest.mark.parametrize( "model", [ - "databricks/databricks-meta-llama-3-1-70b-instruct", + "databricks/databricks-meta-llama-3.2-3b-instruct", "databricks/databricks-meta-llama-3-70b-instruct", "databricks/databricks-dbrx-instruct", # "databricks/databricks-mixtral-8x7b-instruct",