feature: TF242 ioc support (aws#2982)

Co-authored-by: Mufaddal Rohawala <[email protected]>

Author: 2 people

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": {

src/sagemaker/model.py

@@ -630,6 +630,8 @@ def _compilation_job_config(
 
         if (
             framework.lower() == "pytorch"
+            or framework.lower() == "tensorflow"
+            and target_instance_type is not None
             and re.match("(?=^ml_)(?!ml_inf)", target_instance_type) is not None
             and framework_version is not None
         ):