Skip to content
Merged
Show file tree
Hide file tree
Changes from all 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
1 change: 1 addition & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,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)

Expand Down
8 changes: 2 additions & 6 deletions src/sagemaker/amazon/amazon_estimator.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -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)
Expand Down