mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-24 18:24:20 +00:00
test: move drop_params inside test
This commit is contained in:
parent
9593df23c4
commit
a6dd3c0bf5
1 changed files with 1 additions and 2 deletions
|
@ -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"}],
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue