feature: Adding support for 1P Algorithms in HYD

secastillo · secastillo · commit a2ac3917bf71 · 2023-03-15T19:50:08.000-04:00
diff --git a/src/sagemaker/image_uri_config/blazingtext.json b/src/sagemaker/image_uri_config/blazingtext.json
@@ -10,6 +10,7 @@
                 "ap-northeast-3": "867004704886",
                 "ap-southeast-3": "951798379941",
                 "ap-south-1": "991648021394",
+                "ap-south-2": "628508329040",
                 "ap-southeast-1": "475088953585",
                 "ap-southeast-2": "544295431143",
                 "ca-central-1": "469771592824",
@@ -31,6 +32,7 @@
                 "us-gov-west-1": "226302683700",
                 "us-gov-east-1": "237065988967",
                 "us-iso-east-1": "490574956308",
+                "us-isob-east-1": "765400339828",
                 "us-west-1": "632365934929",
                 "us-west-2": "433757028032"
             },
diff --git a/src/sagemaker/image_uri_config/factorization-machines.json b/src/sagemaker/image_uri_config/factorization-machines.json
@@ -10,6 +10,7 @@
                 "ap-northeast-3": "867004704886",
                 "ap-southeast-3": "951798379941",
                 "ap-south-1": "991648021394",
+                "ap-south-2": "628508329040",
                 "ap-southeast-1": "475088953585",
                 "ap-southeast-2": "712309505854",
                 "ca-central-1": "469771592824",
@@ -31,6 +32,7 @@
                 "us-gov-west-1": "226302683700",
                 "us-gov-east-1": "237065988967",
                 "us-iso-east-1": "490574956308",
+                "us-isob-east-1": "765400339828",
                 "us-west-1": "632365934929",
                 "us-west-2": "174872318107"
             },
diff --git a/src/sagemaker/image_uri_config/image-classification.json b/src/sagemaker/image_uri_config/image-classification.json
@@ -0,0 +1,42 @@
+{
+    "scope": ["inference", "training"],
+    "versions": {
+        "1": {
+            "registries": {
+                "af-south-1": "455444449433",
+                "ap-east-1": "286214385809",
+                "ap-northeast-1": "501404015308",
+                "ap-northeast-2": "306986355934",
+                "ap-south-1": "991648021394",
+                "ap-south-2": "628508329040",
+                "ap-southeast-1": "475088953585",
+                "ap-southeast-2": "544295431143",
+                "ap-southeast-3": "951798379941",
+                "ap-northeast-3": "867004704886",
+                "ca-central-1": "469771592824",
+                "cn-north-1": "390948362332",
+                "cn-northwest-1": "387376663083",
+                "eu-central-1": "813361260812",
+                "eu-central-2": "680994064768",
+                "eu-north-1": "669576153137",
+                "eu-west-1": "685385470294",
+                "eu-west-2": "644912444149",
+                "eu-west-3": "749696950732",
+                "eu-south-1": "257386234256",
+                "eu-south-2": "104374241257",
+                "me-south-1": "249704162688",
+                "me-central-1": "272398656194",
+                "sa-east-1": "855470959533",
+                "us-east-1": "811284229777",
+                "us-east-2": "825641698319",
+                "us-gov-west-1": "226302683700",
+                "us-gov-east-1": "237065988967",
+                "us-iso-east-1": "490574956308",
+                "us-isob-east-1": "765400339828",
+                "us-west-1": "632365934929",
+                "us-west-2": "433757028032"
+            },
+            "repository": "image-classification"
+        }
+    }
+}
diff --git a/src/sagemaker/image_uri_config/ipinsights.json b/src/sagemaker/image_uri_config/ipinsights.json
@@ -9,6 +9,7 @@
                 "ap-northeast-2": "835164637446",
                 "ap-northeast-3": "867004704886",
                 "ap-south-1": "991648021394",
+                "ap-south-2": "628508329040",
                 "ap-southeast-1": "475088953585",
                 "ap-southeast-2": "712309505854",
                 "ap-southeast-3": "951798379941",
@@ -31,6 +32,7 @@
                 "us-gov-west-1": "226302683700",
                 "us-gov-east-1": "237065988967",
                 "us-iso-east-1": "490574956308",
+                "us-isob-east-1": "765400339828",
                 "us-west-1": "632365934929",
                 "us-west-2": "174872318107"
             },
diff --git a/src/sagemaker/image_uri_config/kmeans.json b/src/sagemaker/image_uri_config/kmeans.json
@@ -9,6 +9,7 @@
                 "ap-northeast-2": "835164637446",
                 "ap-northeast-3": "867004704886",
                 "ap-south-1": "991648021394",
+                "ap-south-2": "628508329040",
                 "ap-southeast-1": "475088953585",
                 "ap-southeast-2": "712309505854",
                 "ap-southeast-3": "951798379941",
@@ -31,6 +32,7 @@
                 "us-gov-west-1": "226302683700",
                 "us-gov-east-1": "237065988967",
                 "us-iso-east-1": "490574956308",
+                "us-isob-east-1": "765400339828",
                 "us-west-1": "632365934929",
                 "us-west-2": "174872318107"
             },
diff --git a/src/sagemaker/image_uri_config/knn.json b/src/sagemaker/image_uri_config/knn.json
@@ -9,6 +9,7 @@
                 "ap-northeast-2": "835164637446",
                 "ap-northeast-3": "867004704886",
                 "ap-south-1": "991648021394",
+                "ap-south-2": "628508329040",
                 "ap-southeast-1": "475088953585",
                 "ap-southeast-2": "712309505854",
                 "ap-southeast-3": "951798379941",
@@ -31,6 +32,7 @@
                 "us-gov-west-1": "226302683700",
                 "us-gov-east-1": "237065988967",
                 "us-iso-east-1": "490574956308",
+                "us-isob-east-1": "765400339828",
                 "us-west-1": "632365934929",
                 "us-west-2": "174872318107"
             },
diff --git a/src/sagemaker/image_uri_config/linear-learner.json b/src/sagemaker/image_uri_config/linear-learner.json
@@ -9,6 +9,7 @@
                 "ap-northeast-2": "835164637446",
                 "ap-northeast-3": "867004704886",
                 "ap-south-1": "991648021394",
+                "ap-south-2": "628508329040",
                 "ap-southeast-1": "475088953585",
                 "ap-southeast-2": "712309505854",
                 "ap-southeast-3": "951798379941",
@@ -31,6 +32,7 @@
                 "us-gov-west-1": "226302683700",
                 "us-gov-east-1": "237065988967",
                 "us-iso-east-1": "490574956308",
+                "us-isob-east-1": "765400339828",
                 "us-west-1": "632365934929",
                 "us-west-2": "174872318107"
             },
diff --git a/src/sagemaker/image_uri_config/ntm.json b/src/sagemaker/image_uri_config/ntm.json
@@ -9,6 +9,7 @@
                 "ap-northeast-2": "835164637446",
                 "ap-northeast-3": "867004704886",
                 "ap-south-1": "991648021394",
+                "ap-south-2": "628508329040",
                 "ap-southeast-1": "475088953585",
                 "ap-southeast-2": "712309505854",
                 "ap-southeast-3": "951798379941",
@@ -31,6 +32,7 @@
                 "us-gov-west-1": "226302683700",
                 "us-gov-east-1": "237065988967",
                 "us-iso-east-1": "490574956308",
+                "us-isob-east-1": "765400339828",
                 "us-west-1": "632365934929",
                 "us-west-2": "174872318107"
             },
diff --git a/src/sagemaker/image_uri_config/object-detection.json b/src/sagemaker/image_uri_config/object-detection.json
@@ -9,6 +9,7 @@
                 "ap-northeast-2": "306986355934",
                 "ap-northeast-3": "867004704886",
                 "ap-south-1": "991648021394",
+                "ap-south-2": "628508329040",
                 "ap-southeast-1": "475088953585",
                 "ap-southeast-2": "544295431143",
                 "ap-southeast-3": "951798379941",
@@ -31,6 +32,7 @@
                 "us-gov-west-1": "226302683700",
                 "us-gov-east-1": "237065988967",
                 "us-iso-east-1": "490574956308",
+                "us-isob-east-1": "765400339828",
                 "us-west-1": "632365934929",
                 "us-west-2": "433757028032"
             },
diff --git a/src/sagemaker/image_uri_config/object2vec.json b/src/sagemaker/image_uri_config/object2vec.json
@@ -9,6 +9,7 @@
                 "ap-northeast-2": "835164637446",
                 "ap-northeast-3": "867004704886",
                 "ap-south-1": "991648021394",
+                "ap-south-2": "628508329040",
                 "ap-southeast-1": "475088953585",
                 "ap-southeast-2": "712309505854",
                 "ap-southeast-3": "951798379941",
@@ -31,6 +32,7 @@
                 "us-gov-west-1": "226302683700",
                 "us-gov-east-1": "237065988967",
                 "us-iso-east-1": "490574956308",
+                "us-isob-east-1": "765400339828",
                 "us-west-1": "632365934929",
                 "us-west-2": "174872318107"
             },
diff --git a/src/sagemaker/image_uri_config/pca.json b/src/sagemaker/image_uri_config/pca.json
@@ -9,6 +9,7 @@
                 "ap-northeast-2": "835164637446",
                 "ap-northeast-3": "867004704886",
                 "ap-south-1": "991648021394",
+                "ap-south-2": "628508329040",
                 "ap-southeast-1": "475088953585",
                 "ap-southeast-2": "712309505854",
                 "ap-southeast-3": "951798379941",
@@ -31,6 +32,7 @@
                 "us-gov-west-1": "226302683700",
                 "us-gov-east-1": "237065988967",
                 "us-iso-east-1": "490574956308",
+                "us-isob-east-1": "765400339828",
                 "us-west-1": "632365934929",
                 "us-west-2": "174872318107"
             },
diff --git a/src/sagemaker/image_uri_config/randomcutforest.json b/src/sagemaker/image_uri_config/randomcutforest.json
@@ -9,6 +9,7 @@
                 "ap-northeast-2": "835164637446",
                 "ap-northeast-3": "867004704886",
                 "ap-south-1": "991648021394",
+                "ap-south-2": "628508329040",
                 "ap-southeast-1": "475088953585",
                 "ap-southeast-2": "712309505854",
                 "ap-southeast-3": "951798379941",
@@ -31,6 +32,7 @@
                 "us-gov-west-1": "226302683700",
                 "us-gov-east-1": "237065988967",
                 "us-iso-east-1": "490574956308",
+                "us-isob-east-1": "765400339828",
                 "us-west-1": "632365934929",
                 "us-west-2": "174872318107"
             },
diff --git a/src/sagemaker/image_uri_config/semantic-segmentation.json b/src/sagemaker/image_uri_config/semantic-segmentation.json
@@ -9,6 +9,7 @@
                 "ap-northeast-2": "306986355934",
                 "ap-northeast-3": "867004704886",
                 "ap-south-1": "991648021394",
+                "ap-south-2": "628508329040",
                 "ap-southeast-1": "475088953585",
                 "ap-southeast-2": "544295431143",
                 "ap-southeast-3": "951798379941",
@@ -31,6 +32,7 @@
                 "us-gov-west-1": "226302683700",
                 "us-gov-east-1": "237065988967",
                 "us-iso-east-1": "490574956308",
+                "us-isob-east-1": "765400339828",
                 "us-west-1": "632365934929",
                 "us-west-2": "433757028032"
             },
diff --git a/src/sagemaker/image_uri_config/seq2seq.json b/src/sagemaker/image_uri_config/seq2seq.json
@@ -9,6 +9,7 @@
                 "ap-northeast-2": "306986355934",
                 "ap-northeast-3": "867004704886",
                 "ap-south-1": "991648021394",
+                "ap-south-2": "628508329040",
                 "ap-southeast-1": "475088953585",
                 "ap-southeast-2": "544295431143",
                 "ap-southeast-3": "951798379941",
@@ -31,6 +32,7 @@
                 "us-gov-west-1": "226302683700",
                 "us-gov-east-1": "237065988967",
                 "us-iso-east-1": "490574956308",
+                "us-isob-east-1": "765400339828",
                 "us-west-1": "632365934929",
                 "us-west-2": "433757028032"
             },
diff --git a/src/sagemaker/image_uri_config/sklearn.json b/src/sagemaker/image_uri_config/sklearn.json
@@ -11,6 +11,7 @@
                     "ap-northeast-2": "366743142698",
                     "ap-northeast-3": "867004704886",
                     "ap-south-1": "720646828776",
+                    "ap-south-2": "628508329040",
                     "ap-southeast-1": "121021644041",
                     "ap-southeast-2": "783357654285",
                     "ap-southeast-3": "951798379941",
@@ -33,6 +34,7 @@
                     "us-gov-west-1": "414596584902",
                     "us-gov-east-1": "237065988967",
                     "us-iso-east-1": "833128469047",
+                    "us-isob-east-1": "281123927165",
                     "us-west-1": "746614075791",
                     "us-west-2": "246618743249"
                 },
@@ -48,6 +50,7 @@
                     "ap-northeast-2": "366743142698",
                     "ap-northeast-3": "867004704886",
                     "ap-south-1": "720646828776",
+                    "ap-south-2": "628508329040",
                     "ap-southeast-1": "121021644041",
                     "ap-southeast-2": "783357654285",
                     "ap-southeast-3": "951798379941",
@@ -70,6 +73,7 @@
                     "us-gov-west-1": "414596584902",
                     "us-gov-east-1": "237065988967",
                     "us-iso-east-1": "833128469047",
+                    "us-isob-east-1": "281123927165",
                     "us-west-1": "746614075791",
                     "us-west-2": "246618743249"
                 },
@@ -85,6 +89,7 @@
                     "ap-northeast-2": "366743142698",
                     "ap-northeast-3": "867004704886",
                     "ap-south-1": "720646828776",
+                    "ap-south-2": "628508329040",
                     "ap-southeast-1": "121021644041",
                     "ap-southeast-2": "783357654285",
                     "ap-southeast-3": "951798379941",
@@ -107,6 +112,7 @@
                     "us-gov-west-1": "414596584902",
                     "us-gov-east-1": "237065988967",
                     "us-iso-east-1": "833128469047",
+                    "us-isob-east-1": "281123927165",
                     "us-west-1": "746614075791",
                     "us-west-2": "246618743249"
                 },
@@ -126,6 +132,7 @@
                     "ap-northeast-2": "366743142698",
                     "ap-northeast-3": "867004704886",
                     "ap-south-1": "720646828776",
+                    "ap-south-2": "628508329040",
                     "ap-southeast-1": "121021644041",
                     "ap-southeast-2": "783357654285",
                     "ap-southeast-3": "951798379941",
@@ -148,6 +155,7 @@
                     "us-gov-west-1": "414596584902",
                     "us-gov-east-1": "237065988967",
                     "us-iso-east-1": "833128469047",
+                    "us-isob-east-1": "281123927165",
                     "us-west-1": "746614075791",
                     "us-west-2": "246618743249"
                 },
@@ -163,6 +171,7 @@
                     "ap-northeast-2": "366743142698",
                     "ap-northeast-3": "867004704886",
                     "ap-south-1": "720646828776",
+                    "ap-south-2": "628508329040",
                     "ap-southeast-1": "121021644041",
                     "ap-southeast-2": "783357654285",
                     "ap-southeast-3": "951798379941",
@@ -185,6 +194,7 @@
                     "us-gov-west-1": "414596584902",
                     "us-gov-east-1": "237065988967",
                     "us-iso-east-1": "833128469047",
+                    "us-isob-east-1": "281123927165",
                     "us-west-1": "746614075791",
                     "us-west-2": "246618743249"
                 },
@@ -200,6 +210,7 @@
                     "ap-northeast-2": "366743142698",
                     "ap-northeast-3": "867004704886",
                     "ap-south-1": "720646828776",
+                    "ap-south-2": "628508329040",
                     "ap-southeast-1": "121021644041",
                     "ap-southeast-2": "783357654285",
                     "ap-southeast-3": "951798379941",
@@ -222,6 +233,7 @@
                     "us-gov-west-1": "414596584902",
                     "us-gov-east-1": "237065988967",
                     "us-iso-east-1": "833128469047",
+                    "us-isob-east-1": "281123927165",
                     "us-west-1": "746614075791",
                     "us-west-2": "246618743249"
                 },
@@ -241,6 +253,7 @@
                     "ap-northeast-2": "366743142698",
                     "ap-northeast-3": "867004704886",
                     "ap-south-1": "720646828776",
+                    "ap-south-2": "628508329040",
                     "ap-southeast-1": "121021644041",
                     "ap-southeast-2": "783357654285",
                     "ap-southeast-3": "951798379941",
@@ -263,6 +276,7 @@
                     "us-gov-west-1": "414596584902",
                     "us-gov-east-1": "237065988967",
                     "us-iso-east-1": "833128469047",
+                    "us-isob-east-1": "281123927165",
                     "us-west-1": "746614075791",
                     "us-west-2": "246618743249"
                 },
diff --git a/src/sagemaker/image_uri_config/xgboost.json b/src/sagemaker/image_uri_config/xgboost.json
diff --git a/tests/unit/sagemaker/image_uris/test_algos.py b/tests/unit/sagemaker/image_uris/test_algos.py
diff --git a/tests/unit/sagemaker/image_uris/test_sklearn.py b/tests/unit/sagemaker/image_uris/test_sklearn.py
diff --git a/tests/unit/sagemaker/image_uris/test_xgboost.py b/tests/unit/sagemaker/image_uris/test_xgboost.py
