From 0d374fb7c0807923980d537d1b2962958409ae28 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 15 Aug 2024 18:23:41 -0700
Subject: [PATCH] fix sagemaker test

---
 litellm/llms/sagemaker.py       | 2 +-
 litellm/tests/test_sagemaker.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/litellm/llms/sagemaker.py b/litellm/llms/sagemaker.py
index dab8cb1a30..e3a58a7675 100644
--- a/litellm/llms/sagemaker.py
+++ b/litellm/llms/sagemaker.py
@@ -387,7 +387,7 @@ class SagemakerLLM(BaseAWSLLM):
             original_response=completion_response,
             additional_args={"complete_input_dict": _data},
         )
-        print_verbose(f"raw model_response: {response}")
+        print_verbose(f"raw model_response: {completion_response}")
         ## RESPONSE OBJECT
         try:
             if isinstance(completion_response, list):
diff --git a/litellm/tests/test_sagemaker.py b/litellm/tests/test_sagemaker.py
index 155b07be06..a06a238a38 100644
--- a/litellm/tests/test_sagemaker.py
+++ b/litellm/tests/test_sagemaker.py
@@ -84,7 +84,7 @@ async def test_completion_sagemaker(sync_mode):
 
 
 @pytest.mark.asyncio()
-@pytest.mark.parametrize("sync_mode", [True])
+@pytest.mark.parametrize("sync_mode", [False, True])
 async def test_completion_sagemaker_stream(sync_mode):
     try:
         litellm.set_verbose = False