using the generic template for model monitor image retrieve

Author: Chuyang Deng

src/sagemaker/image_uri_config/model-monitor.json

@@ -1,7 +1,7 @@
 {
     "scope": ["monitoring"],
     "versions": {
-        "1": {
+        "": {
             "registries": {
                 "eu-north-1": "895015795356",
                 "me-south-1": "607024016150",

src/sagemaker/image_uris.py

@@ -21,8 +21,7 @@
 
 logger = logging.getLogger(__name__)
 
-ECR_URI_TEMPLATE = "{registry}.dkr.{hostname}/{repository}:{tag}"
-MONITOR_URI_TEMPLATE = "{registry}.dkr.{hostname}/{repository}"
+ECR_URI_TEMPLATE = "{registry}.dkr.{hostname}/{repository}"
 
 
 def retrieve(
@@ -71,16 +70,15 @@ def retrieve(
 
     repo = version_config["repository"]
 
-    # model-monitoring image uri does not have tags
-    if framework == "model-monitor":
-        return MONITOR_URI_TEMPLATE.format(registry=registry, hostname=hostname, repository=repo)
-
     processor = _processor(
         instance_type, config.get("processors") or version_config.get("processors")
     )
     tag = _format_tag(version_config.get("tag_prefix", version), processor, py_version)
 
-    return ECR_URI_TEMPLATE.format(registry=registry, hostname=hostname, repository=repo, tag=tag)
+    if tag:
+        repo += ":{}".format(tag)
+
+    return ECR_URI_TEMPLATE.format(registry=registry, hostname=hostname, repository=repo)
 
 
 def _config_for_framework_and_scope(framework, image_scope, accelerator_type=None):