From 7a4e0ac6f651b2cc6d4e0fcefd8614a5c9e3d38a Mon Sep 17 00:00:00 2001 From: Philipp Schmid <32632186+philschmid@users.noreply.github.com> Date: Wed, 8 Dec 2021 09:26:41 +0100 Subject: [PATCH] Update test_models_from_hub.py --- tests/integ/test_models_from_hub.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/integ/test_models_from_hub.py b/tests/integ/test_models_from_hub.py index 3125c1c..8b09ce5 100644 --- a/tests/integ/test_models_from_hub.py +++ b/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(