Skip to content
Merged
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 6 additions & 3 deletions src/sagemaker/amazon/amazon_estimator.py
Original file line number Diff line number Diff line change
Expand Up @@ -287,7 +287,8 @@ def registry(region_name, algorithm=None):
"eu-central-1": "664544806723",
"ap-northeast-1": "351501993468",
"ap-northeast-2": "835164637446",
"ap-southeast-2": "712309505854"
"ap-southeast-2": "712309505854",
"us-gov-west-1": "226302683700"
}[region_name]
elif algorithm in ["lda"]:
account_id = {
Expand All @@ -309,7 +310,8 @@ def registry(region_name, algorithm=None):
"eu-central-1": "495149712605",
"ap-northeast-1": "633353088612",
"ap-northeast-2": "204372634319",
"ap-southeast-2": "514117268639"
"ap-southeast-2": "514117268639",
"us-gov-west-1": "226302683700"
}[region_name]
elif algorithm in ["xgboost", "seq2seq", "image-classification", "blazingtext",
"object-detection"]:
Expand All @@ -321,7 +323,8 @@ def registry(region_name, algorithm=None):
"eu-central-1": "813361260812",
"ap-northeast-1": "501404015308",
"ap-northeast-2": "306986355934",
"ap-southeast-2": "544295431143"
"ap-southeast-2": "544295431143",
"us-gov-west-1": "226302683700"
}[region_name]
else:
raise ValueError("Algorithm class:{} doesn't have mapping to account_id with images".format(algorithm))
Expand Down