From b47a14f44e98b3efea80f1b9f8b4b2538e89c864 Mon Sep 17 00:00:00 2001
From: Nadia Yakimakha <32335935+nadiaya@users.noreply.github.com>
Date: Wed, 27 Jun 2018 13:37:30 -0700
Subject: [PATCH] Support SageMaker algorithms in ICN region.

---
 CHANGELOG.rst                            | 5 +++++
 README.rst                               | 2 +-
 setup.py                                 | 2 +-
 src/sagemaker/amazon/amazon_estimator.py | 6 ++++--
 4 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/CHANGELOG.rst b/CHANGELOG.rst
index 8b1d9b0ad5..dc1dcdd19e 100644
--- a/CHANGELOG.rst
+++ b/CHANGELOG.rst
@@ -2,6 +2,11 @@
 CHANGELOG
 =========
 
+1.5.2
+=====
+
+* Support SageMaker algorithms in ICN region
+
 1.5.1
 =====
 
diff --git a/README.rst b/README.rst
index 2ce3d744e8..2159572218 100644
--- a/README.rst
+++ b/README.rst
@@ -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
 ~~~~~~~~~~~~~~~~~~~~~~~~~
diff --git a/setup.py b/setup.py
index 111a5f3fda..cd451134f1 100644
--- a/setup.py
+++ b/setup.py
@@ -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'},
diff --git a/src/sagemaker/amazon/amazon_estimator.py b/src/sagemaker/amazon/amazon_estimator.py
index 6fc94da2f6..08839591f2 100644
--- a/src/sagemaker/amazon/amazon_estimator.py
+++ b/src/sagemaker/amazon/amazon_estimator.py
@@ -276,7 +276,8 @@ 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 = {
@@ -284,7 +285,8 @@ def registry(region_name, algorithm=None):
             "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))