Add multiple Accept types

Author: Balaji Veeramani

src/sagemaker/deserializers.py

@@ -58,7 +58,7 @@ def ACCEPT(self):
 class StringDeserializer(BaseDeserializer):
     """Deserialize data from an inference endpoint into a decoded string."""
 
-    ACCEPT = "application/json"
+    ACCEPT = ["application/json", "text/csv"]
 
     def __init__(self, encoding="UTF-8"):
         """Initialize the string encoding.
@@ -87,7 +87,7 @@ def deserialize(self, stream, content_type):
 class BytesDeserializer(BaseDeserializer):
     """Deserialize a stream of bytes into a bytes object."""
 
-    ACCEPT = "*/*"
+    ACCEPT = ["*/*"]
 
     def deserialize(self, stream, content_type):
         """Read a stream of bytes returned from an inference endpoint.
@@ -108,7 +108,7 @@ def deserialize(self, stream, content_type):
 class CSVDeserializer(BaseDeserializer):
     """Deserialize a stream of bytes into a list of lists."""
 
-    ACCEPT = "text/csv"
+    ACCEPT = ["text/csv"]
 
     def __init__(self, encoding="utf-8"):
         """Initialize the string encoding.
@@ -143,7 +143,7 @@ class StreamDeserializer(BaseDeserializer):
     reading it.
     """
 
-    ACCEPT = "*/*"
+    ACCEPT = ["*/*"]
 
     def deserialize(self, stream, content_type):
         """Returns a stream of the response body and the MIME type of the data.
@@ -161,7 +161,7 @@ def deserialize(self, stream, content_type):
 class NumpyDeserializer(BaseDeserializer):
     """Deserialize a stream of data in the .npy format."""
 
-    ACCEPT = "application/x-npy"
+    ACCEPT = ["application/x-npy"]
 
     def __init__(self, dtype=None, allow_pickle=True):
         """Initialize the dtype and allow_pickle arguments.
@@ -201,7 +201,7 @@ def deserialize(self, stream, content_type):
 class JSONDeserializer(BaseDeserializer):
     """Deserialize JSON data from an inference endpoint into a Python object."""
 
-    ACCEPT = "application/json"
+    ACCEPT = ["application/json"]
 
     def deserialize(self, stream, content_type):
         """Deserialize JSON data from an inference endpoint into a Python object.
@@ -222,7 +222,7 @@ def deserialize(self, stream, content_type):
 class PandasDeserializer(BaseDeserializer):
     """Deserialize CSV or JSON data from an inference endpoint into a pandas dataframe."""
 
-    ACCEPT = "text/csv"
+    ACCEPT = ["text/csv", "application/json"]
 
     def deserialize(self, stream, content_type):
         """Deserialize CSV or JSON data from an inference endpoint into a pandas

src/sagemaker/predictor.py

@@ -131,7 +131,7 @@ def _create_request_args(self, data, initial_args=None, target_model=None, targe
             args["ContentType"] = self.content_type
 
         if self.accept and "Accept" not in args:
-            args["Accept"] = self.accept
+            args["Accept"] = ", ".join(self.accept)
 
         if target_model:
             args["TargetModel"] = target_model