Merge pull request #44 from aws/fix-image-uri

Update test_models_from_hub.py

Author: philschmid

tests/integ/test_models_from_hub.py

@@ -38,8 +38,8 @@ def get_all_ecr_images(registry_id, repository_name, result_key):
     image_tags = [image["imageTag"] for image in images]
     print(image_tags)
     image_regex = re.compile("\d\.\d\.\d-" + device + "-.{4}$")
-    image = sorted(list(filter(image_regex.match, image_tags)), reverse=True)[0]
-    return image
+    tag = sorted(list(filter(image_regex.match, image_tags)), reverse=True)[0]
+    return f"{registry_id}.dkr.ecr.{os.environ.get('AWS_DEFAULT_REGION','us-east-1')}.amazonaws.com/{repository_name}:{tag}"
 
 
 @pytest.mark.parametrize(