Skip to content

Commit bc45bbd

Browse files
pdasamznyangaws
authored andcommitted
Add image URIs for built-in Algorithms for SIN/LHR/BOM/SFO/YUL (#456)
1 parent 3d091b4 commit bc45bbd

File tree

2 files changed

+27
-5
lines changed

2 files changed

+27
-5
lines changed

CHANGELOG.rst

+2-1
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,13 @@ CHANGELOG
33
=========
44

55
========
6-
1.13.dev
6+
1.13.0
77
========
88

99
* feature: Estimator: add input mode to training channels
1010
* feature: Estimator: add model_uri and model_channel_name parameters
1111
* enhancement: Local Mode: support output_path. Can be either file:// or s3://
12+
* enhancement: Added image uris for SageMaker built-in algorithms for SIN/LHR/BOM/SFO/YUL
1213

1314
1.12.0
1415
======

src/sagemaker/amazon/amazon_estimator.py

+25-4
Original file line numberDiff line numberDiff line change
@@ -290,7 +290,12 @@ def registry(region_name, algorithm=None):
290290
"ap-northeast-1": "351501993468",
291291
"ap-northeast-2": "835164637446",
292292
"ap-southeast-2": "712309505854",
293-
"us-gov-west-1": "226302683700"
293+
"us-gov-west-1": "226302683700",
294+
"ap-southeast-1": "475088953585",
295+
"ap-south-1": "991648021394",
296+
"ca-central-1": "469771592824",
297+
"eu-west-2": "644912444149",
298+
"us-west-1": "632365934929",
294299
}[region_name]
295300
elif algorithm in ["lda"]:
296301
account_id = {
@@ -301,7 +306,13 @@ def registry(region_name, algorithm=None):
301306
"eu-central-1": "353608530281",
302307
"ap-northeast-1": "258307448986",
303308
"ap-northeast-2": "293181348795",
304-
"ap-southeast-2": "297031611018"
309+
"ap-southeast-2": "297031611018",
310+
"us-gov-west-1": "226302683700",
311+
"ap-southeast-1": "475088953585",
312+
"ap-south-1": "991648021394",
313+
"ca-central-1": "469771592824",
314+
"eu-west-2": "644912444149",
315+
"us-west-1": "632365934929",
305316
}[region_name]
306317
elif algorithm in ["forecasting-deepar"]:
307318
account_id = {
@@ -313,7 +324,12 @@ def registry(region_name, algorithm=None):
313324
"ap-northeast-1": "633353088612",
314325
"ap-northeast-2": "204372634319",
315326
"ap-southeast-2": "514117268639",
316-
"us-gov-west-1": "226302683700"
327+
"us-gov-west-1": "226302683700",
328+
"ap-southeast-1": "475088953585",
329+
"ap-south-1": "991648021394",
330+
"ca-central-1": "469771592824",
331+
"eu-west-2": "644912444149",
332+
"us-west-1": "632365934929",
317333
}[region_name]
318334
elif algorithm in ["xgboost", "seq2seq", "image-classification", "blazingtext",
319335
"object-detection"]:
@@ -326,7 +342,12 @@ def registry(region_name, algorithm=None):
326342
"ap-northeast-1": "501404015308",
327343
"ap-northeast-2": "306986355934",
328344
"ap-southeast-2": "544295431143",
329-
"us-gov-west-1": "226302683700"
345+
"us-gov-west-1": "226302683700",
346+
"ap-southeast-1": "475088953585",
347+
"ap-south-1": "991648021394",
348+
"ca-central-1": "469771592824",
349+
"eu-west-2": "644912444149",
350+
"us-west-1": "632365934929",
330351
}[region_name]
331352
else:
332353
raise ValueError("Algorithm class:{} doesn't have mapping to account_id with images".format(algorithm))

0 commit comments

Comments
 (0)