Skip to content

Commit 1591f82

Browse files
committed
fix: Update localmode code to decode urllib response as UTF8
1 parent b7996b9 commit 1591f82

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/sagemaker/local/entities.py

Lines changed: 4 additions & 3 deletions
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)
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:
@@ -643,8 +643,9 @@ def _perform_request(endpoint_url, pool_manager=None):
643643
"""Placeholder docstring."""
644644
http = pool_manager or urllib3.PoolManager()
645645
try:
646-
r = http.request("GET", endpoint_url)
647-
code = r.status
646+
ret = http.request("GET", endpoint_url)
647+
r = ret.data.decode("utf-8")
648+
code = ret.status
648649
except urllib3.exceptions.RequestError:
649650
return None, -1
650651
return r, code

0 commit comments

Comments
 (0)