diff --git a/src/sagemaker/image_uri_config/blazingtext.json b/src/sagemaker/image_uri_config/blazingtext.json index c94c9cf214..2d6287ff64 100644 --- a/src/sagemaker/image_uri_config/blazingtext.json +++ b/src/sagemaker/image_uri_config/blazingtext.json @@ -3,6 +3,7 @@ "versions": { "1": { "registries": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "501404015308", "ap-northeast-2": "306986355934", @@ -17,6 +18,7 @@ "eu-west-1": "685385470294", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "811284229777", diff --git a/src/sagemaker/image_uri_config/factorization-machines.json b/src/sagemaker/image_uri_config/factorization-machines.json index 56f425c0f3..05aa170bd6 100644 --- a/src/sagemaker/image_uri_config/factorization-machines.json +++ b/src/sagemaker/image_uri_config/factorization-machines.json @@ -3,6 +3,7 @@ "versions": { "1": { "registries": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "351501993468", "ap-northeast-2": "835164637446", @@ -17,6 +18,7 @@ "eu-west-1": "438346466558", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "382416733822", diff --git a/src/sagemaker/image_uri_config/forecasting-deepar.json b/src/sagemaker/image_uri_config/forecasting-deepar.json index dae3a2b2c3..4524887dee 100644 --- a/src/sagemaker/image_uri_config/forecasting-deepar.json +++ b/src/sagemaker/image_uri_config/forecasting-deepar.json @@ -3,6 +3,7 @@ "versions": { "1": { "registries": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "633353088612", "ap-northeast-2": "204372634319", @@ -17,6 +18,7 @@ "eu-west-1": "224300973850", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "522234722520", diff --git a/src/sagemaker/image_uri_config/image-classification.json b/src/sagemaker/image_uri_config/image-classification.json index b5b9591965..5f7d2a6f87 100644 --- a/src/sagemaker/image_uri_config/image-classification.json +++ b/src/sagemaker/image_uri_config/image-classification.json @@ -3,6 +3,7 @@ "versions": { "1": { "registries": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "501404015308", "ap-northeast-2": "306986355934", @@ -17,6 +18,7 @@ "eu-west-1": "685385470294", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "811284229777", diff --git a/src/sagemaker/image_uri_config/ipinsights.json b/src/sagemaker/image_uri_config/ipinsights.json index c1638a0277..4812c512a7 100644 --- a/src/sagemaker/image_uri_config/ipinsights.json +++ b/src/sagemaker/image_uri_config/ipinsights.json @@ -3,6 +3,7 @@ "versions": { "1": { "registries": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "351501993468", "ap-northeast-2": "835164637446", @@ -17,6 +18,7 @@ "eu-west-1": "438346466558", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "382416733822", diff --git a/src/sagemaker/image_uri_config/kmeans.json b/src/sagemaker/image_uri_config/kmeans.json index 457936d91b..7cdee133a5 100644 --- a/src/sagemaker/image_uri_config/kmeans.json +++ b/src/sagemaker/image_uri_config/kmeans.json @@ -3,6 +3,7 @@ "versions": { "1": { "registries": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "351501993468", "ap-northeast-2": "835164637446", @@ -17,6 +18,7 @@ "eu-west-1": "438346466558", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "382416733822", diff --git a/src/sagemaker/image_uri_config/knn.json b/src/sagemaker/image_uri_config/knn.json index d8596f1a52..5b032d45b0 100644 --- a/src/sagemaker/image_uri_config/knn.json +++ b/src/sagemaker/image_uri_config/knn.json @@ -3,6 +3,7 @@ "versions": { "1": { "registries": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "351501993468", "ap-northeast-2": "835164637446", @@ -17,6 +18,7 @@ "eu-west-1": "438346466558", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "382416733822", diff --git a/src/sagemaker/image_uri_config/linear-learner.json b/src/sagemaker/image_uri_config/linear-learner.json index ec0ae2ca10..83fc177557 100644 --- a/src/sagemaker/image_uri_config/linear-learner.json +++ b/src/sagemaker/image_uri_config/linear-learner.json @@ -3,6 +3,7 @@ "versions": { "1": { "registries": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "351501993468", "ap-northeast-2": "835164637446", @@ -17,6 +18,7 @@ "eu-west-1": "438346466558", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "382416733822", diff --git a/src/sagemaker/image_uri_config/ntm.json b/src/sagemaker/image_uri_config/ntm.json index 78fcbdf2fc..86d4a72be4 100644 --- a/src/sagemaker/image_uri_config/ntm.json +++ b/src/sagemaker/image_uri_config/ntm.json @@ -3,6 +3,7 @@ "versions": { "1": { "registries": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "351501993468", "ap-northeast-2": "835164637446", @@ -17,6 +18,7 @@ "eu-west-1": "438346466558", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "382416733822", diff --git a/src/sagemaker/image_uri_config/object-detection.json b/src/sagemaker/image_uri_config/object-detection.json index a439f1043b..5c63dcfc7c 100644 --- a/src/sagemaker/image_uri_config/object-detection.json +++ b/src/sagemaker/image_uri_config/object-detection.json @@ -3,6 +3,7 @@ "versions": { "1": { "registries": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "501404015308", "ap-northeast-2": "306986355934", @@ -17,6 +18,7 @@ "eu-west-1": "685385470294", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "811284229777", diff --git a/src/sagemaker/image_uri_config/object2vec.json b/src/sagemaker/image_uri_config/object2vec.json index 2052db7cc5..c451cba681 100644 --- a/src/sagemaker/image_uri_config/object2vec.json +++ b/src/sagemaker/image_uri_config/object2vec.json @@ -3,6 +3,7 @@ "versions": { "1": { "registries": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "351501993468", "ap-northeast-2": "835164637446", @@ -17,6 +18,7 @@ "eu-west-1": "438346466558", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "382416733822", diff --git a/src/sagemaker/image_uri_config/pca.json b/src/sagemaker/image_uri_config/pca.json index 652a0ae99d..887e9e8a45 100644 --- a/src/sagemaker/image_uri_config/pca.json +++ b/src/sagemaker/image_uri_config/pca.json @@ -3,6 +3,7 @@ "versions": { "1": { "registries": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "351501993468", "ap-northeast-2": "835164637446", @@ -17,6 +18,7 @@ "eu-west-1": "438346466558", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "382416733822", diff --git a/src/sagemaker/image_uri_config/randomcutforest.json b/src/sagemaker/image_uri_config/randomcutforest.json index 05fadc0188..1ab304825e 100644 --- a/src/sagemaker/image_uri_config/randomcutforest.json +++ b/src/sagemaker/image_uri_config/randomcutforest.json @@ -3,6 +3,7 @@ "versions": { "1": { "registries": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "351501993468", "ap-northeast-2": "835164637446", @@ -17,6 +18,7 @@ "eu-west-1": "438346466558", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "382416733822", diff --git a/src/sagemaker/image_uri_config/semantic-segmentation.json b/src/sagemaker/image_uri_config/semantic-segmentation.json index d0dc33df6d..5bc852fb11 100644 --- a/src/sagemaker/image_uri_config/semantic-segmentation.json +++ b/src/sagemaker/image_uri_config/semantic-segmentation.json @@ -3,6 +3,7 @@ "versions": { "1": { "registries": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "501404015308", "ap-northeast-2": "306986355934", @@ -17,6 +18,7 @@ "eu-west-1": "685385470294", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "811284229777", diff --git a/src/sagemaker/image_uri_config/seq2seq.json b/src/sagemaker/image_uri_config/seq2seq.json index 73f4c2dcd6..9515d53f0e 100644 --- a/src/sagemaker/image_uri_config/seq2seq.json +++ b/src/sagemaker/image_uri_config/seq2seq.json @@ -3,6 +3,7 @@ "versions": { "1": { "registries": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "501404015308", "ap-northeast-2": "306986355934", @@ -17,6 +18,7 @@ "eu-west-1": "685385470294", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "811284229777", diff --git a/src/sagemaker/image_uri_config/sklearn.json b/src/sagemaker/image_uri_config/sklearn.json index 962c2632e6..42b7dbc20e 100644 --- a/src/sagemaker/image_uri_config/sklearn.json +++ b/src/sagemaker/image_uri_config/sklearn.json @@ -5,6 +5,7 @@ "0.20.0": { "py_versions": ["py3"], "registries": { + "af-south-1": "510948584623", "ap-east-1": "651117190479", "ap-northeast-1": "354813040037", "ap-northeast-2": "366743142698", @@ -19,6 +20,7 @@ "eu-west-1": "141502667606", "eu-west-2": "764974769150", "eu-west-3": "659782779980", + "eu-south-1": "978288397137", "me-south-1": "801668240914", "sa-east-1": "737474898029", "us-east-1": "683313688378", diff --git a/src/sagemaker/image_uri_config/sparkml-serving.json b/src/sagemaker/image_uri_config/sparkml-serving.json index e767d75729..de9d0e0d4c 100644 --- a/src/sagemaker/image_uri_config/sparkml-serving.json +++ b/src/sagemaker/image_uri_config/sparkml-serving.json @@ -3,6 +3,7 @@ "versions": { "2.2": { "registries": { + "af-south-1": "510948584623", "ap-east-1": "651117190479", "ap-northeast-1": "354813040037", "ap-northeast-2": "366743142698", @@ -17,6 +18,7 @@ "eu-west-1": "141502667606", "eu-west-2": "764974769150", "eu-west-3": "659782779980", + "eu-south-1": "978288397137", "me-south-1": "801668240914", "sa-east-1": "737474898029", "us-east-1": "683313688378", diff --git a/src/sagemaker/image_uri_config/xgboost.json b/src/sagemaker/image_uri_config/xgboost.json index 246f80cbd0..3bf344910c 100644 --- a/src/sagemaker/image_uri_config/xgboost.json +++ b/src/sagemaker/image_uri_config/xgboost.json @@ -6,6 +6,7 @@ "versions": { "1": { "registries": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "501404015308", "ap-northeast-2": "306986355934", @@ -20,6 +21,7 @@ "eu-west-1": "685385470294", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "811284229777", @@ -35,6 +37,7 @@ "processors": ["cpu"], "py_versions": ["py3"], "registries": { + "af-south-1": "510948584623", "ap-east-1": "651117190479", "ap-northeast-1": "354813040037", "ap-northeast-2": "366743142698", @@ -49,6 +52,7 @@ "eu-west-1": "141502667606", "eu-west-2": "764974769150", "eu-west-3": "659782779980", + "eu-south-1": "978288397137", "me-south-1": "801668240914", "sa-east-1": "737474898029", "us-east-1": "683313688378", @@ -64,6 +68,7 @@ "processors": ["cpu"], "py_versions": ["py3"], "registries": { + "af-south-1": "510948584623", "ap-east-1": "651117190479", "ap-northeast-1": "354813040037", "ap-northeast-2": "366743142698", @@ -78,6 +83,7 @@ "eu-west-1": "141502667606", "eu-west-2": "764974769150", "eu-west-3": "659782779980", + "eu-south-1": "978288397137", "me-south-1": "801668240914", "sa-east-1": "737474898029", "us-east-1": "683313688378", @@ -93,6 +99,7 @@ "processors": ["cpu"], "py_versions": ["py3"], "registries": { + "af-south-1": "510948584623", "ap-east-1": "651117190479", "ap-northeast-1": "354813040037", "ap-northeast-2": "366743142698", @@ -107,6 +114,7 @@ "eu-west-1": "141502667606", "eu-west-2": "764974769150", "eu-west-3": "659782779980", + "eu-south-1": "978288397137", "me-south-1": "801668240914", "sa-east-1": "737474898029", "us-east-1": "683313688378", diff --git a/tests/unit/sagemaker/image_uris/test_algos.py b/tests/unit/sagemaker/image_uris/test_algos.py index 493a7dba01..e051f53933 100644 --- a/tests/unit/sagemaker/image_uris/test_algos.py +++ b/tests/unit/sagemaker/image_uris/test_algos.py @@ -48,6 +48,7 @@ "ipinsights", ), "accounts": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "351501993468", "ap-northeast-2": "835164637446", @@ -62,6 +63,7 @@ "eu-west-1": "438346466558", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "382416733822", @@ -95,6 +97,7 @@ { "algorithms": ("forecasting-deepar",), "accounts": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "633353088612", "ap-northeast-2": "204372634319", @@ -109,6 +112,7 @@ "eu-west-1": "224300973850", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "522234722520", @@ -128,6 +132,7 @@ "semantic-segmentation", ), "accounts": { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "501404015308", "ap-northeast-2": "306986355934", @@ -142,6 +147,7 @@ "eu-west-1": "685385470294", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "811284229777", diff --git a/tests/unit/sagemaker/image_uris/test_sklearn.py b/tests/unit/sagemaker/image_uris/test_sklearn.py index 75b213d158..97fb5b8a1e 100644 --- a/tests/unit/sagemaker/image_uris/test_sklearn.py +++ b/tests/unit/sagemaker/image_uris/test_sklearn.py @@ -18,6 +18,7 @@ from tests.unit.sagemaker.image_uris import expected_uris, regions ACCOUNTS = { + "af-south-1": "510948584623", "ap-east-1": "651117190479", "ap-northeast-1": "354813040037", "ap-northeast-2": "366743142698", @@ -32,6 +33,7 @@ "eu-west-1": "141502667606", "eu-west-2": "764974769150", "eu-west-3": "659782779980", + "eu-south-1": "978288397137", "me-south-1": "801668240914", "sa-east-1": "737474898029", "us-east-1": "683313688378", diff --git a/tests/unit/sagemaker/image_uris/test_sparkml.py b/tests/unit/sagemaker/image_uris/test_sparkml.py index e1859f5fab..c9b55584bd 100644 --- a/tests/unit/sagemaker/image_uris/test_sparkml.py +++ b/tests/unit/sagemaker/image_uris/test_sparkml.py @@ -16,6 +16,7 @@ from tests.unit.sagemaker.image_uris import expected_uris, regions ACCOUNTS = { + "af-south-1": "510948584623", "ap-east-1": "651117190479", "ap-northeast-1": "354813040037", "ap-northeast-2": "366743142698", @@ -30,6 +31,7 @@ "eu-west-1": "141502667606", "eu-west-2": "764974769150", "eu-west-3": "659782779980", + "eu-south-1": "978288397137", "me-south-1": "801668240914", "sa-east-1": "737474898029", "us-east-1": "683313688378", diff --git a/tests/unit/sagemaker/image_uris/test_xgboost.py b/tests/unit/sagemaker/image_uris/test_xgboost.py index 7887cb2d44..698e4931af 100644 --- a/tests/unit/sagemaker/image_uris/test_xgboost.py +++ b/tests/unit/sagemaker/image_uris/test_xgboost.py @@ -18,6 +18,7 @@ from tests.unit.sagemaker.image_uris import expected_uris, regions ALGO_REGISTRIES = { + "af-south-1": "455444449433", "ap-east-1": "286214385809", "ap-northeast-1": "501404015308", "ap-northeast-2": "306986355934", @@ -32,6 +33,7 @@ "eu-west-1": "685385470294", "eu-west-2": "644912444149", "eu-west-3": "749696950732", + "eu-south-1": "257386234256", "me-south-1": "249704162688", "sa-east-1": "855470959533", "us-east-1": "811284229777", @@ -42,8 +44,10 @@ "us-west-2": "433757028032", } ALGO_VERSIONS = ("1", "latest") +XGBOOST_FRAMEWORK_VERSIONS = ("0.90-2", "0.90-1", "1.0-1") FRAMEWORK_REGISTRIES = { + "af-south-1": "510948584623", "ap-east-1": "651117190479", "ap-northeast-1": "354813040037", "ap-northeast-2": "366743142698", @@ -58,6 +62,7 @@ "eu-west-1": "141502667606", "eu-west-2": "764974769150", "eu-west-3": "659782779980", + "eu-south-1": "978288397137", "me-south-1": "801668240914", "sa-east-1": "737474898029", "us-east-1": "683313688378", @@ -69,6 +74,7 @@ } +@pytest.mark.parametrize("xgboost_framework_version", XGBOOST_FRAMEWORK_VERSIONS) def test_xgboost_framework(xgboost_framework_version): for region in regions.regions(): uri = image_uris.retrieve( @@ -85,7 +91,7 @@ def test_xgboost_framework(xgboost_framework_version): assert expected == uri -@pytest.mark.parametrize("xgboost_algo_version", ("1", "latest")) +@pytest.mark.parametrize("xgboost_algo_version", ALGO_VERSIONS) def test_xgboost_algo(xgboost_algo_version): for region in regions.regions(): uri = image_uris.retrieve(framework="xgboost", region=region, version=xgboost_algo_version)