diff --git a/src/sagemaker/image_uri_config/image-classification-neo.json b/src/sagemaker/image_uri_config/image-classification-neo.json index b668744f3b..371bd445a9 100644 --- a/src/sagemaker/image_uri_config/image-classification-neo.json +++ b/src/sagemaker/image_uri_config/image-classification-neo.json @@ -25,6 +25,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, diff --git a/src/sagemaker/image_uri_config/inferentia-mxnet.json b/src/sagemaker/image_uri_config/inferentia-mxnet.json index cc8d17cf66..d53c21f7e2 100644 --- a/src/sagemaker/image_uri_config/inferentia-mxnet.json +++ b/src/sagemaker/image_uri_config/inferentia-mxnet.json @@ -27,6 +27,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, @@ -57,6 +59,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, diff --git a/src/sagemaker/image_uri_config/inferentia-pytorch.json b/src/sagemaker/image_uri_config/inferentia-pytorch.json index 6259c7becd..b3d8c80c21 100644 --- a/src/sagemaker/image_uri_config/inferentia-pytorch.json +++ b/src/sagemaker/image_uri_config/inferentia-pytorch.json @@ -27,6 +27,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, @@ -57,6 +59,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, @@ -87,6 +91,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, diff --git a/src/sagemaker/image_uri_config/inferentia-tensorflow.json b/src/sagemaker/image_uri_config/inferentia-tensorflow.json index ae1aa1fd11..ce1d2603b6 100644 --- a/src/sagemaker/image_uri_config/inferentia-tensorflow.json +++ b/src/sagemaker/image_uri_config/inferentia-tensorflow.json @@ -27,6 +27,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, @@ -57,6 +59,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, diff --git a/src/sagemaker/image_uri_config/neo-mxnet.json b/src/sagemaker/image_uri_config/neo-mxnet.json index 66f2453af5..268abde298 100644 --- a/src/sagemaker/image_uri_config/neo-mxnet.json +++ b/src/sagemaker/image_uri_config/neo-mxnet.json @@ -41,6 +41,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, diff --git a/src/sagemaker/image_uri_config/neo-pytorch.json b/src/sagemaker/image_uri_config/neo-pytorch.json index a0fd287132..bd15a6450e 100644 --- a/src/sagemaker/image_uri_config/neo-pytorch.json +++ b/src/sagemaker/image_uri_config/neo-pytorch.json @@ -39,6 +39,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, @@ -69,6 +71,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, @@ -99,6 +103,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, @@ -129,6 +135,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, @@ -159,6 +167,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, diff --git a/src/sagemaker/image_uri_config/neo-tensorflow.json b/src/sagemaker/image_uri_config/neo-tensorflow.json index e5acd35e51..f7fb0fc5c3 100644 --- a/src/sagemaker/image_uri_config/neo-tensorflow.json +++ b/src/sagemaker/image_uri_config/neo-tensorflow.json @@ -47,6 +47,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, @@ -77,6 +79,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, diff --git a/src/sagemaker/image_uri_config/xgboost-neo.json b/src/sagemaker/image_uri_config/xgboost-neo.json index 434c882977..8c44ff381c 100644 --- a/src/sagemaker/image_uri_config/xgboost-neo.json +++ b/src/sagemaker/image_uri_config/xgboost-neo.json @@ -25,6 +25,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009" }, diff --git a/tests/unit/sagemaker/image_uris/expected_uris.py b/tests/unit/sagemaker/image_uris/expected_uris.py index 0574536754..a38ca2c8bf 100644 --- a/tests/unit/sagemaker/image_uris/expected_uris.py +++ b/tests/unit/sagemaker/image_uris/expected_uris.py @@ -16,6 +16,7 @@ "cn-north-1": "amazonaws.com.cn", "cn-northwest-1": "amazonaws.com.cn", "us-iso-east-1": "c2s.ic.gov", + "us-isob-east-1": "sc2s.sgov.gov", } DOMAIN = "amazonaws.com" IMAGE_URI_FORMAT = "{}.dkr.ecr.{}.{}/{}:{}" diff --git a/tests/unit/sagemaker/image_uris/test_neo.py b/tests/unit/sagemaker/image_uris/test_neo.py index d630862381..61d758b403 100644 --- a/tests/unit/sagemaker/image_uris/test_neo.py +++ b/tests/unit/sagemaker/image_uris/test_neo.py @@ -42,6 +42,8 @@ "us-east-1": "785573368785", "us-east-2": "007439368137", "us-gov-west-1": "263933020539", + "us-iso-east-1": "167761179201", + "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009", }