add unit tests for tensorflow and update version for pytorch. Added all supported regions for integration tests

Author: Rohit Kumar Srivastava

tests/integ/__init__.py

@@ -158,6 +158,26 @@
 ]
 # TODO: SM Training Compiler team to add all supported regions.
 TRAINING_COMPILER_SUPPORTED_REGIONS = [
+    "af-south-1",
+    "ap-east-1",
+    "ap-northeast-1",
+    "ap-northeast-2",
+    "ap-northeast-3",
+    "ap-south-1",
+    "ap-southeast-1",
+    "ap-southeast-2",
+    "ca-central-1",
+    "eu-central-1",
+    "eu-north-1",
+    "eu-south-1",
+    "eu-west-1",
+    "eu-west-2",
+    "eu-west-3",
+    "me-south-1",
+    "sa-east-1",
+    "us-east-1",
+    "us-east-2",
+    "us-west-1",
     "us-west-2",
 ]
 # Data parallelism need to be tested with p3.16xlarge.

tests/unit/sagemaker/training_compiler/test_huggingface_compiler.py renamed to tests/unit/sagemaker/training_compiler/test_huggingface_pytorch_compiler.py

@@ -158,9 +158,9 @@ def test_unsupported_BYOC(
 ):
     byoc = (
         "1.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-trcomp-training:"
-        "1.9.0-"
-        "transformers4.10.2-gpu-"
-        "py38-cu111-ubuntu20.04"
+        "1.10.2-"
+        "transformers4.17.0-gpu-"
+        "py38-cu113-ubuntu20.04"
     )
     with pytest.raises(ValueError):
         HuggingFace(
@@ -451,9 +451,9 @@ def test_attach(
 ):
     training_image = (
         "1.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-trcomp-training:"
-        "1.9.0-"
-        "transformers4.10.2-gpu-"
-        "py38-cu111-ubuntu20.04"
+        "1.10.2-"
+        "transformers4.17.0-gpu-"
+        "py38-cu113-ubuntu20.04"
     )
     returned_job_description = {
         "AlgorithmSpecification": {"TrainingInputMode": "File", "TrainingImage": training_image},
@@ -488,8 +488,8 @@ def test_attach(
     estimator = HuggingFace.attach(training_job_name="hopper", sagemaker_session=sagemaker_session)
     assert estimator.latest_training_job.job_name == "hopper"
     assert estimator.py_version == "py38"
-    assert estimator.framework_version == "4.10.2"
-    assert estimator.pytorch_version == "1.9.0"
+    assert estimator.framework_version == "4.17.0"
+    assert estimator.pytorch_version == "1.10.2"
     assert estimator.role == "arn:aws:iam::366:role/SageMakerRole"
     assert estimator.instance_count == 1
     assert estimator.max_run == 24 * 60 * 60