add 1p algorithm change

Chuyang Deng · Chuyang Deng · commit 022c487d977b · 2019-10-21T15:26:57.000-07:00
diff --git a/src/sagemaker/amazon/amazon_estimator.py b/src/sagemaker/amazon/amazon_estimator.py
@@ -465,6 +465,7 @@ def registry(region_name, algorithm=None):
             "eu-north-1": "669576153137",
             "eu-west-3": "749696950732",
             "sa-east-1": "855470959533",
+            "me-south-1": "249704162688",
         }[region_name]
     elif algorithm in ["lda"]:
         account_id = {
@@ -505,6 +506,7 @@ def registry(region_name, algorithm=None):
             "eu-north-1": "669576153137",
             "eu-west-3": "749696950732",
             "sa-east-1": "855470959533",
+            "me-south-1": "249704162688",
         }[region_name]
     elif algorithm in [
         "xgboost",
@@ -534,6 +536,7 @@ def registry(region_name, algorithm=None):
             "eu-north-1": "669576153137",
             "eu-west-3": "749696950732",
             "sa-east-1": "855470959533",
+            "me-south-1": "249704162688",
         }[region_name]
     elif algorithm in ["image-classification-neo", "xgboost-neo"]:
         account_id = NEO_IMAGE_ACCOUNT[region_name]
diff --git a/src/sagemaker/fw_registry.py b/src/sagemaker/fw_registry.py
@@ -112,6 +112,11 @@
         "scikit-learn": "659782779980",
         "xgboost": "659782779980",
     },
+    "me-south-1": {
+        "sparkml-serving": "801668240914",
+        "scikit-learn": "801668240914",
+        "xgboost": "801668240914",
+    },
 }
 
 
diff --git a/src/sagemaker/model.py b/src/sagemaker/model.py
@@ -40,6 +40,9 @@
         "rk3288",
         "rk3399",
         "sbe_c",
+        "aisage",
+        "qcs603",
+        "qcs605",
     ]
 )
 NEO_ALLOWED_FRAMEWORKS = set(["mxnet", "tensorflow", "pytorch", "onnx", "xgboost"])
@@ -62,6 +65,7 @@
     "ap-south-1": "763008648453",
     "sa-east-1": "756306329178",
     "ca-central-1": "464438896020",
+    "me-south-1": "836785723513",
 }
 
 
diff --git a/tests/unit/test_model.py b/tests/unit/test_model.py
@@ -568,6 +568,7 @@ def test_check_neo_region(sagemaker_session, tmpdir):
         "ap-southeast-2",
         "sa-east-1",
         "ca-central-1",
+        "me-south-1",
     ]
     for region_name in ec2_region_list:
         if region_name in neo_support_region:

Original file line number	Diff line number	Diff line change
`@@ -40,6 +40,9 @@`
`40`	`40`	`"rk3288",`
`41`	`41`	`"rk3399",`
`42`	`42`	`"sbe_c",`
	`43`	`+ "aisage",`
	`44`	`+ "qcs603",`
	`45`	`+ "qcs605",`
`43`	`46`	`]`
`44`	`47`	`)`
`45`	`48`	`NEO_ALLOWED_FRAMEWORKS = set(["mxnet", "tensorflow", "pytorch", "onnx", "xgboost"])`
`@@ -62,6 +65,7 @@`
`62`	`65`	`"ap-south-1": "763008648453",`
`63`	`66`	`"sa-east-1": "756306329178",`
`64`	`67`	`"ca-central-1": "464438896020",`
	`68`	`+ "me-south-1": "836785723513",`
`65`	`69`	`}`
`66`	`70`
`67`	`71`
Original file line number	Diff line number	Diff line change
`@@ -568,6 +568,7 @@ def test_check_neo_region(sagemaker_session, tmpdir):`
`568`	`568`	`"ap-southeast-2",`
`569`	`569`	`"sa-east-1",`
`570`	`570`	`"ca-central-1",`
	`571`	`+ "me-south-1",`
`571`	`572`	`]`
`572`	`573`	`for region_name in ec2_region_list:`
`573`	`574`	`if region_name in neo_support_region:`