From d5274efe6210362224d7a776b3dcf146f399b943 Mon Sep 17 00:00:00 2001 From: Andrew Smith <101757907+andjsmi@users.noreply.github.com> Date: Tue, 18 Mar 2025 11:09:52 +1100 Subject: [PATCH] Update test_sagemaker.py to match changed parameters --- tests/local_testing/test_sagemaker.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/local_testing/test_sagemaker.py b/tests/local_testing/test_sagemaker.py index 8438c3c6ba..ba1ab11596 100644 --- a/tests/local_testing/test_sagemaker.py +++ b/tests/local_testing/test_sagemaker.py @@ -265,7 +265,7 @@ async def test_acompletion_sagemaker_non_stream(): # Assert mock_post.assert_called_once() _, kwargs = mock_post.call_args - args_to_sagemaker = kwargs["json"] + args_to_sagemaker = json.loads(kwargs["data"]) print("Arguments passed to sagemaker=", args_to_sagemaker) assert args_to_sagemaker == expected_payload assert ( @@ -325,7 +325,7 @@ async def test_completion_sagemaker_non_stream(): # Assert mock_post.assert_called_once() _, kwargs = mock_post.call_args - args_to_sagemaker = kwargs["json"] + args_to_sagemaker = json.loads(kwargs["data"]) print("Arguments passed to sagemaker=", args_to_sagemaker) assert args_to_sagemaker == expected_payload assert ( @@ -386,7 +386,7 @@ async def test_completion_sagemaker_prompt_template_non_stream(): # Assert mock_post.assert_called_once() _, kwargs = mock_post.call_args - args_to_sagemaker = kwargs["json"] + args_to_sagemaker = json.loads(kwargs["data"]) print("Arguments passed to sagemaker=", args_to_sagemaker) assert args_to_sagemaker == expected_payload @@ -445,7 +445,7 @@ async def test_completion_sagemaker_non_stream_with_aws_params(): # Assert mock_post.assert_called_once() _, kwargs = mock_post.call_args - args_to_sagemaker = kwargs["json"] + args_to_sagemaker = json.loads(kwargs["data"]) print("Arguments passed to sagemaker=", args_to_sagemaker) assert args_to_sagemaker == expected_payload assert (