Skip to content

Commit 763508c

Browse files
ericangelokimPiali Das
authored and
Piali Das
committed
Added PDT region to amazon estimator. (aws#391)
1 parent 1124993 commit 763508c

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

src/sagemaker/amazon/amazon_estimator.py

+6-3
Original file line numberDiff line numberDiff line change
@@ -287,7 +287,8 @@ def registry(region_name, algorithm=None):
287287
"eu-central-1": "664544806723",
288288
"ap-northeast-1": "351501993468",
289289
"ap-northeast-2": "835164637446",
290-
"ap-southeast-2": "712309505854"
290+
"ap-southeast-2": "712309505854",
291+
"us-gov-west-1": "226302683700"
291292
}[region_name]
292293
elif algorithm in ["lda"]:
293294
account_id = {
@@ -309,7 +310,8 @@ def registry(region_name, algorithm=None):
309310
"eu-central-1": "495149712605",
310311
"ap-northeast-1": "633353088612",
311312
"ap-northeast-2": "204372634319",
312-
"ap-southeast-2": "514117268639"
313+
"ap-southeast-2": "514117268639",
314+
"us-gov-west-1": "226302683700"
313315
}[region_name]
314316
elif algorithm in ["xgboost", "seq2seq", "image-classification", "blazingtext",
315317
"object-detection"]:
@@ -321,7 +323,8 @@ def registry(region_name, algorithm=None):
321323
"eu-central-1": "813361260812",
322324
"ap-northeast-1": "501404015308",
323325
"ap-northeast-2": "306986355934",
324-
"ap-southeast-2": "544295431143"
326+
"ap-southeast-2": "544295431143",
327+
"us-gov-west-1": "226302683700"
325328
}[region_name]
326329
else:
327330
raise ValueError("Algorithm class:{} doesn't have mapping to account_id with images".format(algorithm))

0 commit comments

Comments
 (0)