Skip to content

Commit 807b0ec

Browse files
author
Ignacio Quintero
committed
Fix Unit test :(
1 parent dd9851f commit 807b0ec

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/unit/test_local_entities.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ def test_local_transform_job_perform_batch_inference(get_config_value, move_to_d
149149

150150
runtime_client = Mock()
151151
response_object = Mock()
152-
response_object.read.return_value = 'data'
152+
response_object.read.return_value = b'data'
153153
runtime_client.invoke_endpoint.return_value = {'Body': response_object}
154154
local_transform_job.local_session.sagemaker_runtime_client = runtime_client
155155

0 commit comments

Comments
 (0)