edit frameversion comment and specifier set function

Author: BruceChang9688

src/sagemaker/huggingface/training_compiler/config.py

@@ -107,8 +107,7 @@ def validate(cls, estimator):
                 Version(estimator.pytorch_version) in SpecifierSet("> 1.11")
             ):
                 error_helper_string = (
-                    "Training Compiler is only valid between HuggingFace PyTorch 1.9-1.11 "
-                    "for SageMaker Training Compiler."
+                    "SageMaker Training Compiler is only supported with HuggingFace PyTorch 1.9-1.11 "
                     " Received pytorch_version={} which is unsupported."
                 )
                 raise ValueError(error_helper_string.format(estimator.pytorch_version))

src/sagemaker/tensorflow/training_compiler/config.py

@@ -92,26 +92,17 @@ def validate(cls, estimator):
         super(TrainingCompilerConfig, cls).validate(estimator)
 
         if estimator.framework_version:
-            if Version(estimator.framework_version) in SpecifierSet(
-                f"< {cls.MIN_SUPPORTED_VERSION}"
+            if not Version(estimator.framework_version) in SpecifierSet(
+                f">= {cls.MIN_SUPPORTED_VERSION}", f"<= {cls.MAX_SUPPORTED_VERSION}"
             ):
                 error_helper_string = (
                     "SageMaker Training Compiler only supports TensorFlow version "
-                    ">= {} but received {}"
+                    "between {} to {} but received {}"
                 )
                 error_helper_string = error_helper_string.format(
-                    cls.MIN_SUPPORTED_VERSION, estimator.framework_version
-                )
-                raise ValueError(error_helper_string)
-            if Version(estimator.framework_version) in SpecifierSet(
-                f"> {cls.MAX_SUPPORTED_VERSION}"
-            ):
-                error_helper_string = (
-                    "SageMaker Training Compiler only supports TensorFlow version "
-                    "<= {} but received {}"
-                )
-                error_helper_string = error_helper_string.format(
-                    cls.MAX_SUPPORTED_VERSION, estimator.framework_version
+                    cls.MIN_SUPPORTED_VERSION,
+                    cls.MAX_SUPPORTED_VERSION,
+                    estimator.framework_version,
                 )
                 raise ValueError(error_helper_string)