Skip to content

Commit

Permalink
Support SageMaker algorithms in ICN region. (#262)
Browse files Browse the repository at this point in the history
  • Loading branch information
nadiaya authored Jun 27, 2018
1 parent 143357e commit 7dd06ca
Show file tree
Hide file tree
Showing 4 changed files with 11 additions and 4 deletions.
5 changes: 5 additions & 0 deletions CHANGELOG.rst
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,11 @@
CHANGELOG
=========

1.5.2
=====

* Support SageMaker algorithms in ICN region

1.5.1
=====

Expand Down
2 changes: 1 addition & 1 deletion README.rst
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ You can install from source by cloning this repository and issuing a pip install

git clone https://github.com/aws/sagemaker-python-sdk.git
python setup.py sdist
pip install dist/sagemaker-1.5.1.tar.gz
pip install dist/sagemaker-1.5.2.tar.gz

Supported Python versions
~~~~~~~~~~~~~~~~~~~~~~~~~
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ def read(fname):


setup(name="sagemaker",
version="1.5.1",
version="1.5.2",
description="Open source library for training and deploying models on Amazon SageMaker.",
packages=find_packages('src'),
package_dir={'': 'src'},
Expand Down
6 changes: 4 additions & 2 deletions src/sagemaker/amazon/amazon_estimator.py
Original file line number Diff line number Diff line change
Expand Up @@ -276,15 +276,17 @@ def registry(region_name, algorithm=None):
"us-east-2": "404615174143",
"us-west-2": "174872318107",
"eu-west-1": "438346466558",
"ap-northeast-1": "351501993468"
"ap-northeast-1": "351501993468",
"ap-northeast-2": "835164637446"
}[region_name]
elif algorithm in ["lda"]:
account_id = {
"us-east-1": "766337827248",
"us-east-2": "999911452149",
"us-west-2": "266724342769",
"eu-west-1": "999678624901",
"ap-northeast-1": "258307448986"
"ap-northeast-1": "258307448986",
"ap-northeast-2": "293181348795"
}[region_name]
else:
raise ValueError("Algorithm class:{} doesn't have mapping to account_id with images".format(algorithm))
Expand Down

0 comments on commit 7dd06ca

Please sign in to comment.