fix test failures

Author: tejaschumbalkar

tests/integ/test_training_compiler.py

@@ -224,8 +224,12 @@ def test_tensorflow(
     """
     Test the TensorFlow estimator
     """
-    if version.parse(tensorflow_training_latest_version) < version.parse("2.9"):
-        pytest.skip("Training Compiler only supports TF >= 2.9")
+    if (
+        version.parse("2.9")
+        <= version.parse(tensorflow_training_latest_version)
+        < version.parse("2.12")
+    ):
+        pytest.skip("Training Compiler only supports TF >= 2.9 and < 2.12")
     with timeout(minutes=TRAINING_DEFAULT_TIMEOUT_MINUTES):
         epochs = 10
         batch = 256

tests/unit/sagemaker/image_uris/test_dlc_frameworks.py

@@ -72,7 +72,9 @@ def _test_image_uris(
     }
 
     TYPES_AND_PROCESSORS = INSTANCE_TYPES_AND_PROCESSORS
-    if framework == "pytorch" and Version(fw_version) >= Version("1.13"):
+    if (framework == "pytorch" and Version(fw_version) >= Version("1.13")) or (
+        framework == "tensorflow" and Version(fw_version) >= Version("2.12")
+    ):
         """Handle P2 deprecation"""
         TYPES_AND_PROCESSORS = RENEWED_PYTORCH_INSTANCE_TYPES_AND_PROCESSORS
 

tests/unit/sagemaker/training_compiler/test_tensorflow_compiler.py

@@ -57,8 +57,8 @@
 
 @pytest.fixture(scope="module", autouse=True)
 def skip_if_incompatible(tensorflow_training_version, request):
-    if version.parse(tensorflow_training_version) < version.parse("2.9"):
-        pytest.skip("Training Compiler only supports TF >= 2.9")
+    if version.parse("2.9") <= version.parse(tensorflow_training_version) < version.parse("2.12"):
+        pytest.skip("Training Compiler only supports TF >= 2.9 and < 2.12")
 
 
 @pytest.fixture(scope="module")