fix versioning for trcomp

Author: tejaschumbalkar

tests/integ/test_training_compiler.py

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

tests/unit/sagemaker/training_compiler/test_tensorflow_compiler.py

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