From b18e958ddab7b9fb5bcd00d6fcb210504daf7295 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 11 Feb 2025 19:39:55 -0800 Subject: [PATCH] test: fix test --- tests/llm_translation/test_max_completion_tokens.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/llm_translation/test_max_completion_tokens.py b/tests/llm_translation/test_max_completion_tokens.py index a15934e1342..c20ef3d1a06 100644 --- a/tests/llm_translation/test_max_completion_tokens.py +++ b/tests/llm_translation/test_max_completion_tokens.py @@ -143,7 +143,8 @@ def test_all_model_configs(): ) assert ( - "max_completion_tokens" in VertexAILlama3Config().get_supported_openai_params() + "max_completion_tokens" + in VertexAILlama3Config().get_supported_openai_params(model="llama3") ) assert VertexAILlama3Config().map_openai_params( {"max_completion_tokens": 10}, {}, "llama3", drop_params=False