Skip to content

Commit 32efde6

Browse files
committed
change: Updating Sagemaker Neo to be current with the cli
1 parent 91e658c commit 32efde6

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

src/sagemaker/model.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,15 +23,18 @@
2323

2424
LOGGER = logging.getLogger('sagemaker')
2525

26-
NEO_ALLOWED_TARGET_INSTANCE_FAMILY = set(['ml_c5', 'ml_m5', 'ml_c4', 'ml_m4', 'jetson_tx1', 'jetson_tx2', 'ml_p2',
27-
'ml_p3', 'deeplens', 'rasp3b'])
26+
NEO_ALLOWED_TARGET_INSTANCE_FAMILY = set(['ml_c5', 'ml_m5', 'ml_c4', 'ml_m4',
27+
'jetson_tx1', 'jetson_tx2', 'jetson_nano', 'ml_p2',
28+
'ml_p3', 'deeplens', 'rasp3b',
29+
'rk3288', 'rk3399', 'sbe_c'])
2830
NEO_ALLOWED_FRAMEWORKS = set(['mxnet', 'tensorflow', 'pytorch', 'onnx', 'xgboost'])
2931

3032
NEO_IMAGE_ACCOUNT = {
3133
'us-west-2': '301217895009',
3234
'us-east-1': '785573368785',
3335
'eu-west-1': '802834080501',
34-
'us-east-2': '007439368137'
36+
'us-east-2': '007439368137',
37+
'ap-northeast-1': '941853720454'
3538
}
3639

3740

0 commit comments

Comments
 (0)