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

fix sagemaker test

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-15 18:23:41 -07:00
parent 953a67ba4c
commit 3407f20b31
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/sagemaker.py
View file

@ -387,7 +387,7 @@ class SagemakerLLM(BaseAWSLLM):
original_response=completion_response, original_response=completion_response,
additional_args={"complete_input_dict": _data}, additional_args={"complete_input_dict": _data},
) )
print_verbose(f"raw model_response: {response}") print_verbose(f"raw model_response: {completion_response}")
## RESPONSE OBJECT ## RESPONSE OBJECT
try: try:
if isinstance(completion_response, list): if isinstance(completion_response, list):

2
litellm/tests/test_sagemaker.py
View file

@ -84,7 +84,7 @@ async def test_completion_sagemaker(sync_mode):
@pytest.mark.asyncio() @pytest.mark.asyncio()
@pytest.mark.parametrize("sync_mode", [True]) @pytest.mark.parametrize("sync_mode", [False, True])
async def test_completion_sagemaker_stream(sync_mode): async def test_completion_sagemaker_stream(sync_mode):
try: try:
litellm.set_verbose = False litellm.set_verbose = False