Remove LegacySerializer and LegacyDeserializer

Author: Balaji Veeramani

src/sagemaker/predictor.py

@@ -13,9 +13,7 @@
 """Placeholder docstring"""
 from __future__ import print_function, absolute_import
 
-from sagemaker.deserializers import BaseDeserializer
 from sagemaker.model_monitor import DataCaptureConfig
-from sagemaker.serializers import BaseSerializer
 from sagemaker.session import production_variant, Session
 from sagemaker.utils import name_from_base
 
@@ -64,11 +62,6 @@ def __init__(
             accept (str): The invocation's "Accept", overriding any accept from
                 the deserializer (default: None).
         """
-        if serializer is not None and not isinstance(serializer, BaseSerializer):
-            serializer = LegacySerializer(serializer)
-        if deserializer is not None and not isinstance(deserializer, BaseDeserializer):
-            deserializer = LegacyDeserializer(deserializer)
-
         self.endpoint_name = endpoint_name
         self.sagemaker_session = sagemaker_session or Session()
         self.serializer = serializer
@@ -115,8 +108,6 @@ def _handle_response(self, response):
         """
         response_body = response["Body"]
         if self.deserializer is not None:
-            if not isinstance(self.deserializer, BaseDeserializer):
-                self.deserializer = LegacyDeserializer(self.deserializer)
             # It's the deserializer's responsibility to close the stream
             return self.deserializer.deserialize(response_body, response["ContentType"])
         data = response_body.read()
@@ -149,8 +140,6 @@ def _create_request_args(self, data, initial_args=None, target_model=None, targe
             args["TargetVariant"] = target_variant
 
         if self.serializer is not None:
-            if not isinstance(self.serializer, BaseSerializer):
-                self.serializer = LegacySerializer(self.serializer)
             data = self.serializer.serialize(data)
 
         args["Body"] = data
@@ -403,85 +392,3 @@ def _get_model_names(self):
         )
         production_variants = endpoint_config["ProductionVariants"]
         return [d["ModelName"] for d in production_variants]
-
-
-class LegacySerializer(BaseSerializer):
-    """Wrapper that makes legacy serializers forward compatibile."""
-
-    def __init__(self, serializer):
-        """Initialize a ``LegacySerializer``.
-
-        Args:
-            serializer (callable): A legacy serializer.
-        """
-        self.serializer = serializer
-        self.content_type = getattr(serializer, "content_type", None)
-
-    def __call__(self, *args, **kwargs):
-        """Wraps the call method of the legacy serializer.
-
-        Args:
-            data (object): Data to be serialized.
-
-        Returns:
-            object: Serialized data used for a request.
-        """
-        return self.serializer(*args, **kwargs)
-
-    def serialize(self, data):
-        """Wraps the call method of the legacy serializer.
-
-        Args:
-            data (object): Data to be serialized.
-
-        Returns:
-            object: Serialized data used for a request.
-        """
-        return self.serializer(data)
-
-    @property
-    def CONTENT_TYPE(self):
-        """The MIME type of the data sent to the inference endpoint."""
-        return self.content_type
-
-
-class LegacyDeserializer(BaseDeserializer):
-    """Wrapper that makes legacy deserializers forward compatibile."""
-
-    def __init__(self, deserializer):
-        """Initialize a ``LegacyDeserializer``.
-
-        Args:
-            deserializer (callable): A legacy deserializer.
-        """
-        self.deserializer = deserializer
-        self.accept = getattr(deserializer, "accept", None)
-
-    def __call__(self, *args, **kwargs):
-        """Wraps the call method of the legacy deserializer.
-
-        Args:
-            data (object): Data to be deserialized.
-            content_type (str): The MIME type of the data.
-
-        Returns:
-            object: The data deserialized into an object.
-        """
-        return self.deserializer(*args, **kwargs)
-
-    def deserialize(self, data, content_type):
-        """Wraps the call method of the legacy deserializer.
-
-        Args:
-            data (object): Data to be deserialized.
-            content_type (str): The MIME type of the data.
-
-        Returns:
-            object: The data deserialized into an object.
-        """
-        return self.deserializer(data, content_type)
-
-    @property
-    def ACCEPT(self):
-        """The content type that is expected from the inference endpoint."""
-        return self.accept