From 972d843c0495149525bdc278da5c43fec554e599 Mon Sep 17 00:00:00 2001 From: Edward Cho Date: Mon, 1 Jul 2019 20:48:35 +0000 Subject: [PATCH] fix: Add ap-northeast-1 to Neo algorithms region map --- CHANGELOG.md | 1 + src/sagemaker/amazon/amazon_estimator.py | 8 ++------ 2 files changed, 3 insertions(+), 6 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index a92ea01779..daef167a2b 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -9,6 +9,7 @@ ### Bug fixes and other changes * fix git test in test_estimator.py + * Add ap-northeast-1 to Neo algorithms region map ## v1.31.1 (2019-07-01) diff --git a/src/sagemaker/amazon/amazon_estimator.py b/src/sagemaker/amazon/amazon_estimator.py index 94e1e3b340..2936399b26 100644 --- a/src/sagemaker/amazon/amazon_estimator.py +++ b/src/sagemaker/amazon/amazon_estimator.py @@ -22,6 +22,7 @@ from sagemaker.amazon.hyperparameter import Hyperparameter as hp # noqa from sagemaker.amazon.common import write_numpy_to_dense_tensor from sagemaker.estimator import EstimatorBase, _TrainingJob +from sagemaker.model import NEO_IMAGE_ACCOUNT from sagemaker.session import s3_input from sagemaker.utils import sagemaker_timestamp, get_ecr_image_uri_prefix @@ -395,12 +396,7 @@ def registry(region_name, algorithm=None): "us-iso-east-1": "490574956308", }[region_name] elif algorithm in ["image-classification-neo", "xgboost-neo"]: - account_id = { - "us-west-2": "301217895009", - "us-east-1": "785573368785", - "eu-west-1": "802834080501", - "us-east-2": "007439368137", - }[region_name] + account_id = NEO_IMAGE_ACCOUNT[region_name] else: raise ValueError( "Algorithm class:{} does not have mapping to account_id with images".format(algorithm)