Add ModelBase

Author: bveeramani

src/sagemaker/model.py

@@ -13,6 +13,7 @@
 """Placeholder docstring"""
 from __future__ import absolute_import
 
+import abc
 import json
 import logging
 import os
@@ -29,6 +30,7 @@
     git_utils,
 )
 from sagemaker.deprecations import removed_kwargs
+from sagemaker.predictor import PredictorBase
 from sagemaker.transformer import Transformer
 
 LOGGER = logging.getLogger("sagemaker")
@@ -38,7 +40,23 @@
 )
 
 
-class Model(object):
+class ModelBase(abc.ABC):
+    """An object that encapsulates a trained model.
+
+    Models can be deployed to compute services like a SageMaker ``Endpoint``
+    or Lambda. Deployed models can be used to perform real-time inference.
+    """
+
+    @abc.abstractmethod
+    def deploy(self, *args, **kwargs) -> PredictorBase:
+        """Deploy this model to a compute service."""
+
+    @abc.abstractmethod
+    def destroy(self, *args, **kwargs) -> None:
+        """Destroy resources associated with this model."""
+
+
+class Model(ModelBase):
     """A SageMaker ``Model`` that can be deployed to an ``Endpoint``."""
 
     def __init__(
@@ -878,6 +896,8 @@ def delete_model(self):
             )
         self.sagemaker_session.delete_model(self.name)
 
+    destroy = delete_model
+
 
 SCRIPT_PARAM_NAME = "sagemaker_program"
 DIR_PARAM_NAME = "sagemaker_submit_directory"