update TF242 ioc support

Qingzi-Lan · Qingzi-Lan · commit ea5479324947 · 2022-03-09T12:42:22.000-08:00
diff --git a/src/sagemaker/image_uri_config/neo-tensorflow.json b/src/sagemaker/image_uri_config/neo-tensorflow.json
@@ -13,6 +13,9 @@
         "1.12.0": "1.15.3",
         "1.13.0": "1.15.3",
         "1.14.0": "1.15.3",
+        "2.4": "2.4.2",
+        "2.4.0": "2.4.2",
+        "2.4.1": "2.4.2",
         "2.4.2": "2.4.2"
     },
     "versions": {
diff --git a/src/sagemaker/model.py b/src/sagemaker/model.py
@@ -620,7 +620,7 @@ def _compilation_job_config(
         }
 
         if (
-            framework.lower() == "pytorch"
+            framework.lower() == "pytorch" or framework.lower() == "tensorflow"
             and re.match("(?=^ml_)(?!ml_inf)", target_instance_type) is not None
             and framework_version is not None
         ):

Original file line number	Diff line number	Diff line change
`@@ -620,7 +620,7 @@ def _compilation_job_config(`
`620`	`620`	`}`
`621`	`621`
`622`	`622`	`if (`
`623`		`- framework.lower() == "pytorch"`
	`623`	`+ framework.lower() == "pytorch" or framework.lower() == "tensorflow"`
`624`	`624`	`and re.match("(?=^ml_)(?!ml_inf)", target_instance_type) is not None`
`625`	`625`	`and framework_version is not None`
`626`	`626`	`):`