change: remove NEO_ALLOWED_TARGET_INSTANCE_FAMILY (#1285)

Author: eddiecho

src/sagemaker/estimator.py

@@ -42,7 +42,7 @@
 )
 from sagemaker.job import _Job
 from sagemaker.local import LocalSession
-from sagemaker.model import Model, NEO_ALLOWED_TARGET_INSTANCE_FAMILY, NEO_ALLOWED_FRAMEWORKS
+from sagemaker.model import Model, NEO_ALLOWED_FRAMEWORKS
 from sagemaker.model import (
     SCRIPT_PARAM_NAME,
     DIR_PARAM_NAME,
@@ -511,11 +511,6 @@ def compile_model(
             sagemaker.model.Model: A SageMaker ``Model`` object. See
             :func:`~sagemaker.model.Model` for full details.
         """
-        if target_instance_family not in NEO_ALLOWED_TARGET_INSTANCE_FAMILY:
-            raise ValueError(
-                "Please use valid target_instance_family,"
-                "allowed values: {}".format(NEO_ALLOWED_TARGET_INSTANCE_FAMILY)
-            )
         if framework and framework not in NEO_ALLOWED_FRAMEWORKS:
             raise ValueError(
                 "Please use valid framework, allowed values: {}".format(NEO_ALLOWED_FRAMEWORKS)

src/sagemaker/model.py

@@ -24,27 +24,6 @@
 
 LOGGER = logging.getLogger("sagemaker")
 
-NEO_ALLOWED_TARGET_INSTANCE_FAMILY = set(
-    [
-        "ml_c5",
-        "ml_m5",
-        "ml_c4",
-        "ml_m4",
-        "jetson_tx1",
-        "jetson_tx2",
-        "jetson_nano",
-        "ml_p2",
-        "ml_p3",
-        "deeplens",
-        "rasp3b",
-        "rk3288",
-        "rk3399",
-        "sbe_c",
-        "aisage",
-        "qcs603",
-        "qcs605",
-    ]
-)
 NEO_ALLOWED_FRAMEWORKS = set(["mxnet", "tensorflow", "keras", "pytorch", "onnx", "xgboost"])
 
 NEO_IMAGE_ACCOUNT = {