From c18939508d5dc33dfaf7c464a1c379f1bdf9385d Mon Sep 17 00:00:00 2001 From: Yadan Wei Date: Tue, 4 Feb 2025 09:55:59 -0800 Subject: [PATCH 1/2] change test-ecr-scan image data storage --- test/dlc_tests/sanity/test_ecr_scan.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/test/dlc_tests/sanity/test_ecr_scan.py b/test/dlc_tests/sanity/test_ecr_scan.py index a0753546867a..86a883e76067 100644 --- a/test/dlc_tests/sanity/test_ecr_scan.py +++ b/test/dlc_tests/sanity/test_ecr_scan.py @@ -108,7 +108,10 @@ def upload_json_to_image_data_storage_s3_bucket( ) s3_resource = boto3.resource("s3") sts_client = boto3.client("sts") - account_id = os.getenv("AUTOPATCH_STORAGE_ACCOUNT") or sts_client.get_caller_identity().get( + # account_id = os.getenv("AUTOPATCH_STORAGE_ACCOUNT") or sts_client.get_caller_identity().get( + # "Account" + # ) + account_id = sts_client.get_caller_identity().get( "Account" ) for to_upload in upload_list: From 325e9163dc597959041848e08dd39e31c5bafb17 Mon Sep 17 00:00:00 2001 From: Yadan Wei Date: Wed, 12 Feb 2025 09:54:24 -0800 Subject: [PATCH 2/2] add tf2.18 arm64 inference entry --- release_images_patches.yml | 26 ++++++++++++++++++++++++++ test/dlc_tests/sanity/test_ecr_scan.py | 5 +---- 2 files changed, 27 insertions(+), 4 deletions(-) diff --git a/release_images_patches.yml b/release_images_patches.yml index 69e62d3de1e6..4612e1d03254 100644 --- a/release_images_patches.yml +++ b/release_images_patches.yml @@ -370,3 +370,29 @@ release_images: example: False disable_sm_tag: False force_release: False + 29: + framework: "tensorflow" + version: "2.18.0" + arch_type: "arm64" + customer_type: "ec2" + inference: + device_types: [ "cpu" ] + python_versions: [ "py310" ] + os_version: "ubuntu20.04" + cuda_version: "cu122" + example: False + disable_sm_tag: False + force_release: False + 30: + framework: "tensorflow" + version: "2.18.0" + arch_type: "arm64" + customer_type: "sagemaker" + inference: + device_types: [ "cpu" ] + python_versions: [ "py310" ] + os_version: "ubuntu20.04" + cuda_version: "cu122" + example: False + disable_sm_tag: False + force_release: False diff --git a/test/dlc_tests/sanity/test_ecr_scan.py b/test/dlc_tests/sanity/test_ecr_scan.py index 86a883e76067..a0753546867a 100644 --- a/test/dlc_tests/sanity/test_ecr_scan.py +++ b/test/dlc_tests/sanity/test_ecr_scan.py @@ -108,10 +108,7 @@ def upload_json_to_image_data_storage_s3_bucket( ) s3_resource = boto3.resource("s3") sts_client = boto3.client("sts") - # account_id = os.getenv("AUTOPATCH_STORAGE_ACCOUNT") or sts_client.get_caller_identity().get( - # "Account" - # ) - account_id = sts_client.get_caller_identity().get( + account_id = os.getenv("AUTOPATCH_STORAGE_ACCOUNT") or sts_client.get_caller_identity().get( "Account" ) for to_upload in upload_list: