Skip to content

Commit 8072555

Browse files
author
Balaji Veeramani
committed
Appease integration tests
1 parent cb249f8 commit 8072555

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/sagemaker/predictor.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,8 @@ def _handle_response(self, response):
123123
"""
124124
response_body = response["Body"]
125125
if self.deserializer is not None:
126+
if not isinstance(self.deserializer, BaseDeserializer):
127+
self.deserializer = LegacyDeserializer(self.deserializer)
126128
# It's the deserializer's responsibility to close the stream
127129
return self.deserializer.deserialize(response_body, response["ContentType"])
128130
data = response_body.read()
@@ -155,6 +157,8 @@ def _create_request_args(self, data, initial_args=None, target_model=None, targe
155157
args["TargetVariant"] = target_variant
156158

157159
if self.serializer is not None:
160+
if not isinstance(self.serializer, BaseSerializer):
161+
self.serializer = LegacySerializer(self.serializer)
158162
data = self.serializer.serialize(data)
159163

160164
args["Body"] = data

0 commit comments

Comments
 (0)