feature: Add TLV accounts for 1P Algorithms (#4054)

malav-shastri · Malav Shastri · web-flow · commit 1b3bed392849 · 2023-08-10T20:44:20.000-07:00
Co-authored-by: Malav Shastri &lt;malavhs@amazon.com&gt;
diff --git a/src/sagemaker/image_uri_config/blazingtext.json b/src/sagemaker/image_uri_config/blazingtext.json
@@ -25,6 +25,7 @@
                 "eu-west-3": "749696950732",
                 "eu-south-1": "257386234256",
                 "eu-south-2": "104374241257",
+                "il-central-1": "898809789911",
                 "me-south-1": "249704162688",
                 "me-central-1": "272398656194",
                 "sa-east-1": "855470959533",
diff --git a/src/sagemaker/image_uri_config/factorization-machines.json b/src/sagemaker/image_uri_config/factorization-machines.json
@@ -25,6 +25,7 @@
                 "eu-west-3": "749696950732",
                 "eu-south-1": "257386234256",
                 "eu-south-2": "104374241257",
+                "il-central-1": "898809789911",
                 "me-south-1": "249704162688",
                 "me-central-1": "272398656194",
                 "sa-east-1": "855470959533",
diff --git a/src/sagemaker/image_uri_config/forecasting-deepar.json b/src/sagemaker/image_uri_config/forecasting-deepar.json
@@ -25,6 +25,7 @@
                 "eu-west-3": "749696950732",
                 "eu-south-1": "257386234256",
                 "eu-south-2": "104374241257",
+                "il-central-1": "898809789911",
                 "me-south-1": "249704162688",
                 "me-central-1": "272398656194",
                 "sa-east-1": "855470959533",
diff --git a/src/sagemaker/image_uri_config/image-classification.json b/src/sagemaker/image_uri_config/image-classification.json
@@ -25,6 +25,7 @@
                 "eu-west-3": "749696950732",
                 "eu-south-1": "257386234256",
                 "eu-south-2": "104374241257",
+                "il-central-1": "898809789911",
                 "me-south-1": "249704162688",
                 "me-central-1": "272398656194",
                 "sa-east-1": "855470959533",
diff --git a/src/sagemaker/image_uri_config/ipinsights.json b/src/sagemaker/image_uri_config/ipinsights.json
@@ -25,6 +25,7 @@
                 "eu-west-3": "749696950732",
                 "eu-south-1": "257386234256",
                 "eu-south-2": "104374241257",
+                "il-central-1": "898809789911",
                 "me-south-1": "249704162688",
                 "me-central-1": "272398656194",
                 "sa-east-1": "855470959533",
diff --git a/src/sagemaker/image_uri_config/kmeans.json b/src/sagemaker/image_uri_config/kmeans.json
@@ -25,6 +25,7 @@
                 "eu-west-3": "749696950732",
                 "eu-south-1": "257386234256",
                 "eu-south-2": "104374241257",
+                "il-central-1": "898809789911",
                 "me-south-1": "249704162688",
                 "me-central-1": "272398656194",
                 "sa-east-1": "855470959533",
diff --git a/src/sagemaker/image_uri_config/knn.json b/src/sagemaker/image_uri_config/knn.json
@@ -25,6 +25,7 @@
                 "eu-west-3": "749696950732",
                 "eu-south-1": "257386234256",
                 "eu-south-2": "104374241257",
+                "il-central-1": "898809789911",
                 "me-south-1": "249704162688",
                 "me-central-1": "272398656194",
                 "sa-east-1": "855470959533",
diff --git a/src/sagemaker/image_uri_config/linear-learner.json b/src/sagemaker/image_uri_config/linear-learner.json
@@ -25,6 +25,7 @@
                 "eu-west-3": "749696950732",
                 "eu-south-1": "257386234256",
                 "eu-south-2": "104374241257",
+                "il-central-1": "898809789911",
                 "me-south-1": "249704162688",
                 "me-central-1": "272398656194",
                 "sa-east-1": "855470959533",
diff --git a/src/sagemaker/image_uri_config/ntm.json b/src/sagemaker/image_uri_config/ntm.json
@@ -25,6 +25,7 @@
                 "eu-west-3": "749696950732",
                 "eu-south-1": "257386234256",
                 "eu-south-2": "104374241257",
+                "il-central-1": "898809789911",
                 "me-south-1": "249704162688",
                 "me-central-1": "272398656194",
                 "sa-east-1": "855470959533",
diff --git a/src/sagemaker/image_uri_config/object-detection.json b/src/sagemaker/image_uri_config/object-detection.json
@@ -25,6 +25,7 @@
                 "eu-west-3": "749696950732",
                 "eu-south-1": "257386234256",
                 "eu-south-2": "104374241257",
+                "il-central-1": "898809789911",
                 "me-south-1": "249704162688",
                 "me-central-1": "272398656194",
                 "sa-east-1": "855470959533",
diff --git a/src/sagemaker/image_uri_config/object2vec.json b/src/sagemaker/image_uri_config/object2vec.json
@@ -25,6 +25,7 @@
                 "eu-west-3": "749696950732",
                 "eu-south-1": "257386234256",
                 "eu-south-2": "104374241257",
+                "il-central-1": "898809789911",
                 "me-south-1": "249704162688",
                 "me-central-1": "272398656194",
                 "sa-east-1": "855470959533",
diff --git a/src/sagemaker/image_uri_config/pca.json b/src/sagemaker/image_uri_config/pca.json
@@ -25,6 +25,7 @@
                 "eu-west-3": "749696950732",
                 "eu-south-1": "257386234256",
                 "eu-south-2": "104374241257",
+                "il-central-1": "898809789911",
                 "me-south-1": "249704162688",
                 "me-central-1": "272398656194",
                 "sa-east-1": "855470959533",
diff --git a/src/sagemaker/image_uri_config/randomcutforest.json b/src/sagemaker/image_uri_config/randomcutforest.json
@@ -25,6 +25,7 @@
                 "eu-west-3": "749696950732",
                 "eu-south-1": "257386234256",
                 "eu-south-2": "104374241257",
+                "il-central-1": "898809789911",
                 "me-south-1": "249704162688",
                 "me-central-1": "272398656194",
                 "sa-east-1": "855470959533",
diff --git a/src/sagemaker/image_uri_config/semantic-segmentation.json b/src/sagemaker/image_uri_config/semantic-segmentation.json
@@ -25,6 +25,7 @@
                 "eu-west-3": "749696950732",
                 "eu-south-1": "257386234256",
                 "eu-south-2": "104374241257",
+                "il-central-1": "898809789911",
                 "me-south-1": "249704162688",
                 "me-central-1": "272398656194",
                 "sa-east-1": "855470959533",
diff --git a/src/sagemaker/image_uri_config/seq2seq.json b/src/sagemaker/image_uri_config/seq2seq.json
@@ -25,6 +25,7 @@
                 "eu-west-3": "749696950732",
                 "eu-south-1": "257386234256",
                 "eu-south-2": "104374241257",
+                "il-central-1": "898809789911",
                 "me-south-1": "249704162688",
                 "me-central-1": "272398656194",
                 "sa-east-1": "855470959533",
diff --git a/src/sagemaker/image_uri_config/sklearn.json b/src/sagemaker/image_uri_config/sklearn.json
@@ -27,6 +27,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -67,6 +68,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -107,6 +109,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -147,6 +150,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -191,6 +195,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -231,6 +236,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -271,6 +277,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -311,6 +318,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -355,6 +363,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
diff --git a/src/sagemaker/image_uri_config/xgboost.json b/src/sagemaker/image_uri_config/xgboost.json
@@ -148,6 +148,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -186,6 +187,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -224,6 +226,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -262,6 +265,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -300,6 +304,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -338,6 +343,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -503,6 +509,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -541,6 +548,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -579,6 +587,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -617,6 +626,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -655,6 +665,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -693,6 +704,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -735,6 +747,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
@@ -773,6 +786,7 @@
                     "eu-west-3": "659782779980",
                     "eu-south-1": "978288397137",
                     "eu-south-2": "104374241257",
+                    "il-central-1": "898809789911",
                     "me-south-1": "801668240914",
                     "me-central-1": "272398656194",
                     "sa-east-1": "737474898029",
diff --git a/tests/unit/sagemaker/image_uris/test_algos.py b/tests/unit/sagemaker/image_uris/test_algos.py
@@ -71,6 +71,7 @@
             "eu-west-3": "749696950732",
             "eu-south-1": "257386234256",
             "eu-south-2": "104374241257",
+            "il-central-1": "898809789911",
             "me-south-1": "249704162688",
             "me-central-1": "272398656194",
             "sa-east-1": "855470959533",
@@ -167,6 +168,7 @@
             "eu-west-3": "749696950732",
             "eu-south-1": "257386234256",
             "eu-south-2": "104374241257",
+            "il-central-1": "898809789911",
             "me-south-1": "249704162688",
             "me-central-1": "272398656194",
             "sa-east-1": "855470959533",
diff --git a/tests/unit/sagemaker/image_uris/test_sklearn.py b/tests/unit/sagemaker/image_uris/test_sklearn.py
@@ -40,6 +40,7 @@
     "eu-west-3": "659782779980",
     "eu-south-1": "978288397137",
     "eu-south-2": "104374241257",
+    "il-central-1": "898809789911",
     "me-south-1": "801668240914",
     "me-central-1": "272398656194",
     "sa-east-1": "737474898029",
diff --git a/tests/unit/sagemaker/image_uris/test_xgboost.py b/tests/unit/sagemaker/image_uris/test_xgboost.py
@@ -38,6 +38,7 @@
     "eu-west-3": "749696950732",
     "eu-south-1": "257386234256",
     "eu-south-2": "104374241257",
+    "il-central-1": "898809789911",
     "me-south-1": "249704162688",
     "me-central-1": "272398656194",
     "sa-east-1": "855470959533",
@@ -77,6 +78,7 @@
     "eu-west-3": "659782779980",
     "eu-south-1": "978288397137",
     "eu-south-2": "104374241257",
+    "il-central-1": "898809789911",
     "me-south-1": "801668240914",
     "me-central-1": "272398656194",
     "sa-east-1": "737474898029",
@@ -114,29 +116,33 @@ def test_xgboost_framework(xgboost_framework_version):
 @pytest.mark.parametrize("xgboost_framework_version", XGBOOST_FRAMEWORK_CPU_ONLY_VERSIONS)
 def test_xgboost_framework_cpu_only(xgboost_framework_version):
     for region in FRAMEWORK_REGISTRIES.keys():
-        uri = image_uris.retrieve(
-            framework="xgboost",
-            region=region,
-            version=xgboost_framework_version,
-        )
+        if not (xgboost_framework_version in ["0.90-2", "0.90-1"] and region == "il-central-1"):
+            uri = image_uris.retrieve(
+                framework="xgboost",
+                region=region,
+                version=xgboost_framework_version,
+            )
 
-        expected = expected_uris.framework_uri(
-            "sagemaker-xgboost",
-            xgboost_framework_version,
-            FRAMEWORK_REGISTRIES[region],
-            region=region,
-            py_version="py3",
-            processor="cpu",
-        )
-        assert expected == uri
+            expected = expected_uris.framework_uri(
+                "sagemaker-xgboost",
+                xgboost_framework_version,
+                FRAMEWORK_REGISTRIES[region],
+                region=region,
+                py_version="py3",
+                processor="cpu",
+            )
+            assert expected == uri
 
 
 @pytest.mark.parametrize("xgboost_algo_version", ALGO_VERSIONS)
 def test_xgboost_algo(xgboost_algo_version):
     for region in ALGO_REGISTRIES.keys():
-        uri = image_uris.retrieve(framework="xgboost", region=region, version=xgboost_algo_version)
+        if region != "il-central-1":
+            uri = image_uris.retrieve(
+                framework="xgboost", region=region, version=xgboost_algo_version
+            )
 
-        expected = expected_uris.algo_uri(
-            "xgboost", ALGO_REGISTRIES[region], region, version=xgboost_algo_version
-        )
-        assert expected == uri
+            expected = expected_uris.algo_uri(
+                "xgboost", ALGO_REGISTRIES[region], region, version=xgboost_algo_version
+            )
+            assert expected == uri
