Skip to content

Commit b281592

Browse files
Malav Shastriknikure
Malav Shastri
authored and
knikure
committed
feature: Add TLV accounts for 1P Algorithms
1 parent b8fd4c4 commit b281592

20 files changed

+66
-19
lines changed

‎src/sagemaker/image_uri_config/blazingtext.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
"eu-west-3": "749696950732",
2626
"eu-south-1": "257386234256",
2727
"eu-south-2": "104374241257",
28+
"il-central-1": "898809789911",
2829
"me-south-1": "249704162688",
2930
"me-central-1": "272398656194",
3031
"sa-east-1": "855470959533",

‎src/sagemaker/image_uri_config/factorization-machines.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
"eu-west-3": "749696950732",
2626
"eu-south-1": "257386234256",
2727
"eu-south-2": "104374241257",
28+
"il-central-1": "898809789911",
2829
"me-south-1": "249704162688",
2930
"me-central-1": "272398656194",
3031
"sa-east-1": "855470959533",

‎src/sagemaker/image_uri_config/forecasting-deepar.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
"eu-west-3": "749696950732",
2626
"eu-south-1": "257386234256",
2727
"eu-south-2": "104374241257",
28+
"il-central-1": "898809789911",
2829
"me-south-1": "249704162688",
2930
"me-central-1": "272398656194",
3031
"sa-east-1": "855470959533",

‎src/sagemaker/image_uri_config/image-classification.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
"eu-west-3": "749696950732",
2626
"eu-south-1": "257386234256",
2727
"eu-south-2": "104374241257",
28+
"il-central-1": "898809789911",
2829
"me-south-1": "249704162688",
2930
"me-central-1": "272398656194",
3031
"sa-east-1": "855470959533",

‎src/sagemaker/image_uri_config/ipinsights.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
"eu-west-3": "749696950732",
2626
"eu-south-1": "257386234256",
2727
"eu-south-2": "104374241257",
28+
"il-central-1": "898809789911",
2829
"me-south-1": "249704162688",
2930
"me-central-1": "272398656194",
3031
"sa-east-1": "855470959533",

‎src/sagemaker/image_uri_config/kmeans.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
"eu-west-3": "749696950732",
2626
"eu-south-1": "257386234256",
2727
"eu-south-2": "104374241257",
28+
"il-central-1": "898809789911",
2829
"me-south-1": "249704162688",
2930
"me-central-1": "272398656194",
3031
"sa-east-1": "855470959533",

‎src/sagemaker/image_uri_config/knn.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
"eu-west-3": "749696950732",
2626
"eu-south-1": "257386234256",
2727
"eu-south-2": "104374241257",
28+
"il-central-1": "898809789911",
2829
"me-south-1": "249704162688",
2930
"me-central-1": "272398656194",
3031
"sa-east-1": "855470959533",

‎src/sagemaker/image_uri_config/linear-learner.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
"eu-west-3": "749696950732",
2626
"eu-south-1": "257386234256",
2727
"eu-south-2": "104374241257",
28+
"il-central-1": "898809789911",
2829
"me-south-1": "249704162688",
2930
"me-central-1": "272398656194",
3031
"sa-east-1": "855470959533",

‎src/sagemaker/image_uri_config/ntm.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
"eu-west-3": "749696950732",
2626
"eu-south-1": "257386234256",
2727
"eu-south-2": "104374241257",
28+
"il-central-1": "898809789911",
2829
"me-south-1": "249704162688",
2930
"me-central-1": "272398656194",
3031
"sa-east-1": "855470959533",

‎src/sagemaker/image_uri_config/object-detection.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
"eu-west-3": "749696950732",
2626
"eu-south-1": "257386234256",
2727
"eu-south-2": "104374241257",
28+
"il-central-1": "898809789911",
2829
"me-south-1": "249704162688",
2930
"me-central-1": "272398656194",
3031
"sa-east-1": "855470959533",

‎src/sagemaker/image_uri_config/object2vec.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
"eu-west-3": "749696950732",
2626
"eu-south-1": "257386234256",
2727
"eu-south-2": "104374241257",
28+
"il-central-1": "898809789911",
2829
"me-south-1": "249704162688",
2930
"me-central-1": "272398656194",
3031
"sa-east-1": "855470959533",

‎src/sagemaker/image_uri_config/pca.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
"eu-west-3": "749696950732",
2626
"eu-south-1": "257386234256",
2727
"eu-south-2": "104374241257",
28+
"il-central-1": "898809789911",
2829
"me-south-1": "249704162688",
2930
"me-central-1": "272398656194",
3031
"sa-east-1": "855470959533",

‎src/sagemaker/image_uri_config/randomcutforest.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
"eu-west-3": "749696950732",
2626
"eu-south-1": "257386234256",
2727
"eu-south-2": "104374241257",
28+
"il-central-1": "898809789911",
2829
"me-south-1": "249704162688",
2930
"me-central-1": "272398656194",
3031
"sa-east-1": "855470959533",

‎src/sagemaker/image_uri_config/semantic-segmentation.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
"eu-west-3": "749696950732",
2626
"eu-south-1": "257386234256",
2727
"eu-south-2": "104374241257",
28+
"il-central-1": "898809789911",
2829
"me-south-1": "249704162688",
2930
"me-central-1": "272398656194",
3031
"sa-east-1": "855470959533",

‎src/sagemaker/image_uri_config/seq2seq.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
"eu-west-3": "749696950732",
2626
"eu-south-1": "257386234256",
2727
"eu-south-2": "104374241257",
28+
"il-central-1": "898809789911",
2829
"me-south-1": "249704162688",
2930
"me-central-1": "272398656194",
3031
"sa-east-1": "855470959533",

‎src/sagemaker/image_uri_config/sklearn.json

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
"eu-west-3": "659782779980",
2828
"eu-south-1": "978288397137",
2929
"eu-south-2": "104374241257",
30+
"il-central-1": "898809789911",
3031
"me-south-1": "801668240914",
3132
"me-central-1": "272398656194",
3233
"sa-east-1": "737474898029",
@@ -67,6 +68,7 @@
6768
"eu-west-3": "659782779980",
6869
"eu-south-1": "978288397137",
6970
"eu-south-2": "104374241257",
71+
"il-central-1": "898809789911",
7072
"me-south-1": "801668240914",
7173
"me-central-1": "272398656194",
7274
"sa-east-1": "737474898029",
@@ -107,6 +109,7 @@
107109
"eu-west-3": "659782779980",
108110
"eu-south-1": "978288397137",
109111
"eu-south-2": "104374241257",
112+
"il-central-1": "898809789911",
110113
"me-south-1": "801668240914",
111114
"me-central-1": "272398656194",
112115
"sa-east-1": "737474898029",
@@ -147,6 +150,7 @@
147150
"eu-west-3": "659782779980",
148151
"eu-south-1": "978288397137",
149152
"eu-south-2": "104374241257",
153+
"il-central-1": "898809789911",
150154
"me-south-1": "801668240914",
151155
"me-central-1": "272398656194",
152156
"sa-east-1": "737474898029",
@@ -191,6 +195,7 @@
191195
"eu-west-3": "659782779980",
192196
"eu-south-1": "978288397137",
193197
"eu-south-2": "104374241257",
198+
"il-central-1": "898809789911",
194199
"me-south-1": "801668240914",
195200
"me-central-1": "272398656194",
196201
"sa-east-1": "737474898029",
@@ -231,6 +236,7 @@
231236
"eu-west-3": "659782779980",
232237
"eu-south-1": "978288397137",
233238
"eu-south-2": "104374241257",
239+
"il-central-1": "898809789911",
234240
"me-south-1": "801668240914",
235241
"me-central-1": "272398656194",
236242
"sa-east-1": "737474898029",
@@ -271,6 +277,7 @@
271277
"eu-west-3": "659782779980",
272278
"eu-south-1": "978288397137",
273279
"eu-south-2": "104374241257",
280+
"il-central-1": "898809789911",
274281
"me-south-1": "801668240914",
275282
"me-central-1": "272398656194",
276283
"sa-east-1": "737474898029",
@@ -311,6 +318,7 @@
311318
"eu-west-3": "659782779980",
312319
"eu-south-1": "978288397137",
313320
"eu-south-2": "104374241257",
321+
"il-central-1": "898809789911",
314322
"me-south-1": "801668240914",
315323
"me-central-1": "272398656194",
316324
"sa-east-1": "737474898029",
@@ -355,6 +363,7 @@
355363
"eu-west-3": "659782779980",
356364
"eu-south-1": "978288397137",
357365
"eu-south-2": "104374241257",
366+
"il-central-1": "898809789911",
358367
"me-south-1": "801668240914",
359368
"me-central-1": "272398656194",
360369
"sa-east-1": "737474898029",

‎src/sagemaker/image_uri_config/xgboost.json

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -148,6 +148,7 @@
148148
"eu-west-3": "659782779980",
149149
"eu-south-1": "978288397137",
150150
"eu-south-2": "104374241257",
151+
"il-central-1": "898809789911",
151152
"me-south-1": "801668240914",
152153
"me-central-1": "272398656194",
153154
"sa-east-1": "737474898029",
@@ -186,6 +187,7 @@
186187
"eu-west-3": "659782779980",
187188
"eu-south-1": "978288397137",
188189
"eu-south-2": "104374241257",
190+
"il-central-1": "898809789911",
189191
"me-south-1": "801668240914",
190192
"me-central-1": "272398656194",
191193
"sa-east-1": "737474898029",
@@ -224,6 +226,7 @@
224226
"eu-west-3": "659782779980",
225227
"eu-south-1": "978288397137",
226228
"eu-south-2": "104374241257",
229+
"il-central-1": "898809789911",
227230
"me-south-1": "801668240914",
228231
"me-central-1": "272398656194",
229232
"sa-east-1": "737474898029",
@@ -262,6 +265,7 @@
262265
"eu-west-3": "659782779980",
263266
"eu-south-1": "978288397137",
264267
"eu-south-2": "104374241257",
268+
"il-central-1": "898809789911",
265269
"me-south-1": "801668240914",
266270
"me-central-1": "272398656194",
267271
"sa-east-1": "737474898029",
@@ -300,6 +304,7 @@
300304
"eu-west-3": "659782779980",
301305
"eu-south-1": "978288397137",
302306
"eu-south-2": "104374241257",
307+
"il-central-1": "898809789911",
303308
"me-south-1": "801668240914",
304309
"me-central-1": "272398656194",
305310
"sa-east-1": "737474898029",
@@ -338,6 +343,7 @@
338343
"eu-west-3": "659782779980",
339344
"eu-south-1": "978288397137",
340345
"eu-south-2": "104374241257",
346+
"il-central-1": "898809789911",
341347
"me-south-1": "801668240914",
342348
"me-central-1": "272398656194",
343349
"sa-east-1": "737474898029",
@@ -503,6 +509,7 @@
503509
"eu-west-3": "659782779980",
504510
"eu-south-1": "978288397137",
505511
"eu-south-2": "104374241257",
512+
"il-central-1": "898809789911",
506513
"me-south-1": "801668240914",
507514
"me-central-1": "272398656194",
508515
"sa-east-1": "737474898029",
@@ -541,6 +548,7 @@
541548
"eu-west-3": "659782779980",
542549
"eu-south-1": "978288397137",
543550
"eu-south-2": "104374241257",
551+
"il-central-1": "898809789911",
544552
"me-south-1": "801668240914",
545553
"me-central-1": "272398656194",
546554
"sa-east-1": "737474898029",
@@ -579,6 +587,7 @@
579587
"eu-west-3": "659782779980",
580588
"eu-south-1": "978288397137",
581589
"eu-south-2": "104374241257",
590+
"il-central-1": "898809789911",
582591
"me-south-1": "801668240914",
583592
"me-central-1": "272398656194",
584593
"sa-east-1": "737474898029",
@@ -617,6 +626,7 @@
617626
"eu-west-3": "659782779980",
618627
"eu-south-1": "978288397137",
619628
"eu-south-2": "104374241257",
629+
"il-central-1": "898809789911",
620630
"me-south-1": "801668240914",
621631
"me-central-1": "272398656194",
622632
"sa-east-1": "737474898029",
@@ -655,6 +665,7 @@
655665
"eu-west-3": "659782779980",
656666
"eu-south-1": "978288397137",
657667
"eu-south-2": "104374241257",
668+
"il-central-1": "898809789911",
658669
"me-south-1": "801668240914",
659670
"me-central-1": "272398656194",
660671
"sa-east-1": "737474898029",
@@ -693,6 +704,7 @@
693704
"eu-west-3": "659782779980",
694705
"eu-south-1": "978288397137",
695706
"eu-south-2": "104374241257",
707+
"il-central-1": "898809789911",
696708
"me-south-1": "801668240914",
697709
"me-central-1": "272398656194",
698710
"sa-east-1": "737474898029",
@@ -735,6 +747,7 @@
735747
"eu-west-3": "659782779980",
736748
"eu-south-1": "978288397137",
737749
"eu-south-2": "104374241257",
750+
"il-central-1": "898809789911",
738751
"me-south-1": "801668240914",
739752
"me-central-1": "272398656194",
740753
"sa-east-1": "737474898029",
@@ -773,6 +786,7 @@
773786
"eu-west-3": "659782779980",
774787
"eu-south-1": "978288397137",
775788
"eu-south-2": "104374241257",
789+
"il-central-1": "898809789911",
776790
"me-south-1": "801668240914",
777791
"me-central-1": "272398656194",
778792
"sa-east-1": "737474898029",

‎tests/unit/sagemaker/image_uris/test_algos.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@
7171
"eu-west-3": "749696950732",
7272
"eu-south-1": "257386234256",
7373
"eu-south-2": "104374241257",
74+
"il-central-1": "898809789911",
7475
"me-south-1": "249704162688",
7576
"me-central-1": "272398656194",
7677
"sa-east-1": "855470959533",
@@ -167,6 +168,7 @@
167168
"eu-west-3": "749696950732",
168169
"eu-south-1": "257386234256",
169170
"eu-south-2": "104374241257",
171+
"il-central-1": "898809789911",
170172
"me-south-1": "249704162688",
171173
"me-central-1": "272398656194",
172174
"sa-east-1": "855470959533",

‎tests/unit/sagemaker/image_uris/test_sklearn.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@
4040
"eu-west-3": "659782779980",
4141
"eu-south-1": "978288397137",
4242
"eu-south-2": "104374241257",
43+
"il-central-1": "898809789911",
4344
"me-south-1": "801668240914",
4445
"me-central-1": "272398656194",
4546
"sa-east-1": "737474898029",

‎tests/unit/sagemaker/image_uris/test_xgboost.py

Lines changed: 25 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@
3838
"eu-west-3": "749696950732",
3939
"eu-south-1": "257386234256",
4040
"eu-south-2": "104374241257",
41+
"il-central-1": "898809789911",
4142
"me-south-1": "249704162688",
4243
"me-central-1": "272398656194",
4344
"sa-east-1": "855470959533",
@@ -77,6 +78,7 @@
7778
"eu-west-3": "659782779980",
7879
"eu-south-1": "978288397137",
7980
"eu-south-2": "104374241257",
81+
"il-central-1": "898809789911",
8082
"me-south-1": "801668240914",
8183
"me-central-1": "272398656194",
8284
"sa-east-1": "737474898029",
@@ -114,29 +116,33 @@ def test_xgboost_framework(xgboost_framework_version):
114116
@pytest.mark.parametrize("xgboost_framework_version", XGBOOST_FRAMEWORK_CPU_ONLY_VERSIONS)
115117
def test_xgboost_framework_cpu_only(xgboost_framework_version):
116118
for region in FRAMEWORK_REGISTRIES.keys():
117-
uri = image_uris.retrieve(
118-
framework="xgboost",
119-
region=region,
120-
version=xgboost_framework_version,
121-
)
119+
if not (xgboost_framework_version in ["0.90-2", "0.90-1"] and region == "il-central-1"):
120+
uri = image_uris.retrieve(
121+
framework="xgboost",
122+
region=region,
123+
version=xgboost_framework_version,
124+
)
122125

123-
expected = expected_uris.framework_uri(
124-
"sagemaker-xgboost",
125-
xgboost_framework_version,
126-
FRAMEWORK_REGISTRIES[region],
127-
region=region,
128-
py_version="py3",
129-
processor="cpu",
130-
)
131-
assert expected == uri
126+
expected = expected_uris.framework_uri(
127+
"sagemaker-xgboost",
128+
xgboost_framework_version,
129+
FRAMEWORK_REGISTRIES[region],
130+
region=region,
131+
py_version="py3",
132+
processor="cpu",
133+
)
134+
assert expected == uri
132135

133136

134137
@pytest.mark.parametrize("xgboost_algo_version", ALGO_VERSIONS)
135138
def test_xgboost_algo(xgboost_algo_version):
136139
for region in ALGO_REGISTRIES.keys():
137-
uri = image_uris.retrieve(framework="xgboost", region=region, version=xgboost_algo_version)
140+
if region != "il-central-1":
141+
uri = image_uris.retrieve(
142+
framework="xgboost", region=region, version=xgboost_algo_version
143+
)
138144

139-
expected = expected_uris.algo_uri(
140-
"xgboost", ALGO_REGISTRIES[region], region, version=xgboost_algo_version
141-
)
142-
assert expected == uri
145+
expected = expected_uris.algo_uri(
146+
"xgboost", ALGO_REGISTRIES[region], region, version=xgboost_algo_version
147+
)
148+
assert expected == uri

0 commit comments

Comments
 (0)