Add PredictorBase

Author: bveeramani

src/sagemaker/predictor.py

@@ -13,6 +13,9 @@
 """Placeholder docstring"""
 from __future__ import print_function, absolute_import
 
+import abc
+from typing import Any, Tuple
+
 from sagemaker.deprecations import (
     deprecated_class,
     deprecated_deserialize,
@@ -51,7 +54,29 @@
 from sagemaker.lineage.context import EndpointContext
 
 
-class Predictor(object):
+class PredictorBase(abc.ABC):
+    """An object that encapsulates a deployed model."""
+
+    @abc.abstractmethod
+    def predict(self, *args, **kwargs) -> Any:
+        """Perform inference on the provided data and return a prediction."""
+
+    @abc.abstractmethod
+    def destroy(self, *args, **kwargs) -> None:
+        """Destroy resources associated with this predictor."""
+
+    @property
+    @abc.abstractmethod
+    def content_type(self) -> str:
+        """The MIME type of the data sent to the inference server."""
+
+    @property
+    @abc.abstractmethod
+    def accept(self) -> Tuple[str]:
+        """The content type(s) that are expected from the inference server."""
+
+
+class Predictor(PredictorBase):
     """Make prediction requests to an Amazon SageMaker endpoint."""
 
     def __init__(
@@ -305,6 +330,8 @@ def delete_endpoint(self, delete_endpoint_config=True):
 
         self.sagemaker_session.delete_endpoint(self.endpoint_name)
 
+    destroy = delete_endpoint
+
     def delete_model(self):
         """Deletes the Amazon SageMaker models backing this predictor."""
         request_failed = False