phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test(test_sagemaker.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-23 20:47:56 -07:00
parent 6592de3ef9
commit fd20fbf4fc
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_sagemaker.py
View file

@ -295,7 +295,7 @@ async def test_completion_sagemaker_prompt_template_non_stream():
], ],
temperature=0.2, temperature=0.2,
max_tokens=80, max_tokens=80,
hf_model_name="mistralai/Mistral-7B-Instruct-v0.1", hf_model_name="deepseek-ai/deepseek-coder-6.7b-instruct",
) )
# Print what was called on the mock # Print what was called on the mock