Fix tests

Author: Balaji Veeramani

tests/integ/test_multi_variant_endpoint.py

@@ -173,8 +173,6 @@ def test_predict_invocation_with_target_variant(sagemaker_session, multi_variant
         endpoint_name=multi_variant_endpoint.endpoint_name,
         sagemaker_session=sagemaker_session,
         serializer=CSVSerializer(),
-        content_type="text/csv",
-        accept="text/csv",
     )
 
     # Validate that no exception is raised when the target_variant is specified.

tests/integ/test_tfs.py

@@ -190,10 +190,8 @@ def test_predict_jsons_json_content_type(tfs_predictor):
     predictor = sagemaker.Predictor(
         tfs_predictor.endpoint_name,
         tfs_predictor.sagemaker_session,
-        serializer=None,
+        serializer=sagemaker.serializers.IdentitySerializer(),
         deserializer=sagemaker.deserializers.JSONDeserializer(),
-        content_type="application/json",
-        accept="application/json",
     )
 
     result = predictor.predict(input_data)
@@ -207,10 +205,8 @@ def test_predict_jsons(tfs_predictor):
     predictor = sagemaker.Predictor(
         tfs_predictor.endpoint_name,
         tfs_predictor.sagemaker_session,
-        serializer=None,
+        serializer=sagemaker.serializer.IdentitySerializer(),
         deserializer=sagemaker.deserializers.JSONDeserializer(),
-        content_type="application/jsons",
-        accept="application/jsons",
     )
 
     result = predictor.predict(input_data)
@@ -224,10 +220,8 @@ def test_predict_jsonlines(tfs_predictor):
     predictor = sagemaker.Predictor(
         tfs_predictor.endpoint_name,
         tfs_predictor.sagemaker_session,
-        serializer=None,
-        deserializer=sagemaker.deserializers.JSONDeserializer(),
-        content_type="application/jsonlines",
-        accept="application/jsonlines",
+        serializer=sagemaker.serializers.JSONLinesSerializer(),
+        deserializer=sagemaker.deserializers.JSONLinesDeserializer(),
     )
 
     result = predictor.predict(input_data)

tests/unit/sagemaker/test_serializers.py

@@ -246,6 +246,7 @@ def test_identity_serializer(identity_serializer):
     assert identity_serializer.serialize(b"{}") == b"{}"
 
 
+@pytest.fixture
 def json_lines_serializer():
     return JSONLinesSerializer()