Merge branch 'master' into chore/telemetry-for-deployment-configs

Author: evakravi

src/sagemaker/image_uri_config/pytorch.json

@@ -1008,6 +1008,7 @@
                     "us-gov-west-1": "442386744353",
                     "us-iso-east-1": "886529160074",
                     "us-isob-east-1": "094389454867",
+                    "us-isof-east-1": "303241398832",
                     "us-isof-south-1": "454834333376",
                     "us-west-1": "763104351884",
                     "us-west-2": "763104351884"
@@ -1052,6 +1053,7 @@
                     "us-gov-west-1": "442386744353",
                     "us-iso-east-1": "886529160074",
                     "us-isob-east-1": "094389454867",
+                    "us-isof-east-1": "303241398832",
                     "us-isof-south-1": "454834333376",
                     "us-west-1": "763104351884",
                     "us-west-2": "763104351884"
@@ -2331,6 +2333,7 @@
                     "us-gov-west-1": "442386744353",
                     "us-iso-east-1": "886529160074",
                     "us-isob-east-1": "094389454867",
+                    "us-isof-east-1": "303241398832",
                     "us-isof-south-1": "454834333376",
                     "us-west-1": "763104351884",
                     "us-west-2": "763104351884"
@@ -2419,6 +2422,7 @@
                     "us-gov-west-1": "442386744353",
                     "us-iso-east-1": "886529160074",
                     "us-isob-east-1": "094389454867",
+                    "us-isof-east-1": "303241398832",
                     "us-isof-south-1": "454834333376",
                     "us-west-1": "763104351884",
                     "us-west-2": "763104351884"

src/sagemaker/image_uri_config/tensorflow.json

@@ -2140,6 +2140,7 @@
                     "us-gov-west-1": "442386744353",
                     "us-iso-east-1": "886529160074",
                     "us-isob-east-1": "094389454867",
+                    "us-isof-east-1": "303241398832",
                     "us-isof-south-1": "454834333376",
                     "us-west-1": "763104351884",
                     "us-west-2": "763104351884"
@@ -2181,6 +2182,7 @@
                     "us-gov-west-1": "442386744353",
                     "us-iso-east-1": "886529160074",
                     "us-isob-east-1": "094389454867",
+                    "us-isof-east-1": "303241398832",
                     "us-isof-south-1": "454834333376",
                     "us-west-1": "763104351884",
                     "us-west-2": "763104351884"
@@ -4354,6 +4356,7 @@
                     "us-gov-west-1": "442386744353",
                     "us-iso-east-1": "886529160074",
                     "us-isob-east-1": "094389454867",
+                    "us-isof-east-1": "303241398832",
                     "us-isof-south-1": "454834333376",
                     "us-west-1": "763104351884",
                     "us-west-2": "763104351884"

tests/integ/test_byo_estimator.py

@@ -109,7 +109,7 @@ def test_byo_estimator(sagemaker_session, region, cpu_instance_type, training_se
 
 
 @pytest.mark.release
-def test_estimator_register_publish_training_details(sagemaker_session, region):
+def test_estimator_register_publish_training_details(sagemaker_session, region, cpu_instance_type):
 
     bucket = sagemaker_session.default_bucket()
     prefix = "model-card-sample-notebook"
@@ -150,7 +150,7 @@ def test_estimator_register_publish_training_details(sagemaker_session, region):
         container,
         role="SageMakerRole",
         instance_count=1,
-        instance_type="ml.m4.xlarge",
+        instance_type=cpu_instance_type,
         output_path=output_location,
         sagemaker_session=sagemaker_session,
     )