From a6dd3c0bf5297ec4580cf7fae3cee112fa59140b Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 31 Jan 2024 20:07:28 -0800 Subject: [PATCH] test: move drop_params inside test --- litellm/tests/test_model_max_token_adjust.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/litellm/tests/test_model_max_token_adjust.py b/litellm/tests/test_model_max_token_adjust.py index 8469302a52..b4d48b5e28 100644 --- a/litellm/tests/test_model_max_token_adjust.py +++ b/litellm/tests/test_model_max_token_adjust.py @@ -11,11 +11,10 @@ sys.path.insert( import litellm from litellm import completion -litellm.drop_params = True - def test_completion_sagemaker(): litellm.set_verbose = True + litellm.drop_params = True response = completion( model="sagemaker/berri-benchmarking-Llama-2-70b-chat-hf-4", messages=[{"content": "Hello, how are you?", "role": "user"}],