change: add data wrangler image uri (#2268)

* change: add data wrangler image uri

* fix: add newline and retrigger CI

Co-authored-by: Ajay Karpur <[email protected]>
Co-authored-by: Ahsan Khan <[email protected]>
Co-authored-by: icywang86rui <[email protected]>

Author: 4 people

src/sagemaker/image_uri_config/data-wrangler.json

@@ -0,0 +1,33 @@
+{
+  "processing": {
+    "versions": {
+      "1.x": {
+        "registries": {
+          "af-south-1": "143210264188",
+          "ap-east-1": "707077482487",
+          "ap-northeast-1": "649008135260",
+          "ap-northeast-2": "131546521161",
+          "ap-south-1": "089933028263",
+          "ap-southeast-1": "119527597002",
+          "ap-southeast-2": "422173101802",
+          "ca-central-1": "557239378090",
+          "eu-central-1": "024640144536",
+          "eu-north-1": "054986407534",
+          "eu-south-1": "488287956546",
+          "eu-west-1": "245179582081",
+          "eu-west-2": "894491911112",
+          "eu-west-3": "807237891255",
+          "me-south-1": "376037874950",
+          "sa-east-1": "424196993095",
+          "us-east-1": "663277389841",
+          "us-east-2": "415577184552",
+          "us-west-1": "926135532090",
+          "us-west-2": "174368400705",
+          "cn-north-1": "245909111842",
+          "cn-northwest-1": "249157047649"
+        },
+        "repository": "sagemaker-data-wrangler-container"
+      }
+    }
+  }
+}

tests/unit/sagemaker/image_uris/test_data_wrangler.py

@@ -0,0 +1,55 @@
+# Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You
+# may not use this file except in compliance with the License. A copy of
+# the License is located at
+#
+#     http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file is
+# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
+# ANY KIND, either express or implied. See the License for the specific
+# language governing permissions and limitations under the License.
+from __future__ import absolute_import
+
+from sagemaker import image_uris
+from tests.unit.sagemaker.image_uris import expected_uris, regions
+
+DATA_WRANGLER_ACCOUNTS = {
+    "af-south-1": "143210264188",
+    "ap-east-1": "707077482487",
+    "ap-northeast-1": "649008135260",
+    "ap-northeast-2": "131546521161",
+    "ap-south-1": "089933028263",
+    "ap-southeast-1": "119527597002",
+    "ap-southeast-2": "422173101802",
+    "ca-central-1": "557239378090",
+    "eu-central-1": "024640144536",
+    "eu-north-1": "054986407534",
+    "eu-south-1": "488287956546",
+    "eu-west-1": "245179582081",
+    "eu-west-2": "894491911112",
+    "eu-west-3": "807237891255",
+    "me-south-1": "376037874950",
+    "sa-east-1": "424196993095",
+    "us-east-1": "663277389841",
+    "us-east-2": "415577184552",
+    "us-west-1": "926135532090",
+    "us-west-2": "174368400705",
+    "cn-north-1": "245909111842",
+    "cn-northwest-1": "249157047649",
+}
+
+
+def test_data_wrangler_ecr_uri():
+    for region in regions.regions():
+        if region in DATA_WRANGLER_ACCOUNTS.keys():
+            actual_uri = image_uris.retrieve("data-wrangler", region=region)
+
+            expected_uri = expected_uris.algo_uri(
+                "sagemaker-data-wrangler-container",
+                DATA_WRANGLER_ACCOUNTS[region],
+                region,
+                version="1.x",
+            )
+            assert expected_uri == actual_uri