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"}],