Skip to content

Commit d87749f

Browse files
author
Balaji Veeramani
committed
Fix integration tests
1 parent 3cfcd4a commit d87749f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/integ/test_tfs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -205,7 +205,7 @@ def test_predict_jsons(tfs_predictor):
205205
tfs_predictor.endpoint_name,
206206
tfs_predictor.sagemaker_session,
207207
serializer=None,
208-
deserializer=sagemaker.predictor.JSONDeserializer(),
208+
deserializer=sagemaker.deserializers.JSONDeserializer(),
209209
content_type="application/jsons",
210210
accept="application/jsons",
211211
)
@@ -222,7 +222,7 @@ def test_predict_jsonlines(tfs_predictor):
222222
tfs_predictor.endpoint_name,
223223
tfs_predictor.sagemaker_session,
224224
serializer=None,
225-
deserializer=sagemaker.predictor.JSONDeserializer(),
225+
deserializer=sagemaker.deserializers.JSONDeserializer(),
226226
content_type="application/jsonlines",
227227
accept="application/jsonlines",
228228
)

0 commit comments

Comments
 (0)