fix: update test and version compatibility

Author: Lokiiiiii

tests/unit/sagemaker/training_compiler/test_huggingface_pytorch_compiler.py

@@ -359,6 +359,8 @@ def test_pytorchxla_distribution(
     huggingface_training_compiler_pytorch_version,
     instance_class,
 ):
+    if Version(huggingface_training_compiler_pytorch_version) < Version("1.11"):
+        pytest.skip("This test is intended for PyTorch 1.11 and above")
     compiler_config = TrainingCompilerConfig()
     instance_type = f"ml.{instance_class}.xlarge"
 

tests/unit/sagemaker/training_compiler/test_huggingface_tensorflow_compiler.py

@@ -41,7 +41,7 @@
 ROLE = "Dummy"
 REGION = "us-east-1"
 GPU = "ml.p3.2xlarge"
-SUPPORTED_GPU_INSTANCE_CLASSES = {"p3", "p3dn", "g4dn", "p4dn"}
+SUPPORTED_GPU_INSTANCE_CLASSES = {"p3", "p3dn", "g4dn", "p4d", "g5"}
 UNSUPPORTED_GPU_INSTANCE_CLASSES = EC2_GPU_INSTANCE_CLASSES - SUPPORTED_GPU_INSTANCE_CLASSES
 
 LIST_TAGS_RESULT = {"Tags": [{"Key": "TagtestKey", "Value": "TagtestValue"}]}