test: move drop_params inside test

2024-01-31 20:07:28 -08:00 · 2024-01-31 20:07:28 -08:00 · a6dd3c0bf5
commit a6dd3c0bf5
parent 9593df23c4
1 changed files with 1 additions and 2 deletions
--- 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"}],