change: throw error on TF 1.14 with EIA

Author: mvsusp

src/sagemaker/tensorflow/serving.py

@@ -206,7 +206,7 @@ def deploy(
 
     def _eia_supported(self):
         """Return true if TF version is EIA enabled"""
-        return [int(s) for s in self._framework_version.split(".")] <= self.LATEST_EIA_VERSION
+        return [int(s) for s in self._framework_version.split(".")][:2] <= self.LATEST_EIA_VERSION
 
     def prepare_container_def(self, instance_type, accelerator_type=None):
         """

tests/unit/test_tfs.py

@@ -97,6 +97,19 @@ def test_tfs_model_image_accelerator(sagemaker_session, tf_version):
 
 
 def test_tfs_model_image_accelerator_not_supported(sagemaker_session):
+    model = Model(
+        "s3://some/data.tar.gz",
+        role=ROLE,
+        framework_version="1.13.1",
+        sagemaker_session=sagemaker_session,
+    )
+
+    # assert error is not raised
+
+    model.deploy(
+        instance_type="ml.c4.xlarge", initial_instance_count=1, accelerator_type="ml.eia1.medium"
+    )
+
     model = Model(
         "s3://some/data.tar.gz",
         role=ROLE,