Skip to content

Commit 284ddbe

Browse files
authored
fix: Update localmode code to decode urllib response as UTF8 (#3284)
1 parent b7996b9 commit 284ddbe

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

src/sagemaker/local/entities.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -314,7 +314,7 @@ def start(self, input_data, output_data, transform_resources, **kwargs):
314314
endpoint_url = "http://%s:%d/execution-parameters" % (get_docker_host(), serving_port)
315315
response, code = _perform_request(endpoint_url)
316316
if code == 200:
317-
execution_parameters = json.loads(response.read())
317+
execution_parameters = json.loads(response.data.decode("utf-8"))
318318
# MaxConcurrentTransforms is ignored because we currently only support 1
319319
for setting in ("BatchStrategy", "MaxPayloadInMB"):
320320
if setting not in kwargs and setting in execution_parameters:

tests/unit/test_local_entities.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ def test_start_local_transform_job(_perform_batch_inference, _perform_request, l
106106

107107
response = Mock()
108108
_perform_request.return_value = (response, 200)
109-
response.read.return_value = '{"BatchStrategy": "SingleRecord"}'
109+
response.data = '{"BatchStrategy": "SingleRecord"}'.encode("UTF-8")
110110
local_transform_job.primary_container["ModelDataUrl"] = "file:///some/model"
111111
local_transform_job.start(input_data, output_data, transform_resources, Environment={})
112112

@@ -176,9 +176,9 @@ def test_start_local_transform_job_from_remote_docker_host(
176176
output_data = {}
177177
transform_resources = {"InstanceType": "local"}
178178
m_get_docker_host.return_value = "some_host"
179-
perform_request_mock = Mock()
180-
m_perform_request.return_value = (perform_request_mock, 200)
181-
perform_request_mock.read.return_value = '{"BatchStrategy": "SingleRecord"}'
179+
response = Mock()
180+
m_perform_request.return_value = (response, 200)
181+
response.data = '{"BatchStrategy": "SingleRecord"}'.encode("UTF-8")
182182
local_transform_job.primary_container["ModelDataUrl"] = "file:///some/model"
183183
local_transform_job.start(input_data, output_data, transform_resources, Environment={})
184184
endpoints = [

0 commit comments

Comments
 (0)