updated AML2_CPU_ARM64_US_EAST_1 ami_name_pattern (#5479)

bhanutejagk · Bhanu Teja Goshikonda · web-flow · commit 72c40f42958b · 2025-11-17T09:45:20.000-08:00
* updated AML2_CPU_ARM64_US_EAST_1 ami_name_pattern

* testing change on pytoch-training-2-8-sm

* explicitly declared typer, langcodes and language_data

* reverted toml file

* reverted package declarations for pytorch 2.8 image since tests succeeded for the fix

---------

Co-authored-by: Bhanu Teja Goshikonda &lt;bhanugk@amazon.com&gt;
diff --git a/dlc_developer_config.toml b/dlc_developer_config.toml
@@ -187,4 +187,4 @@ dlc-pr-tensorflow-2-eia-inference = ""
 dlc-pr-vllm = ""
 
 # sglang
-dlc-pr-sglang = ""
+dlc-pr-sglang = ""
diff --git a/test/test_utils/__init__.py b/test/test_utils/__init__.py
@@ -142,7 +142,7 @@ def get_ami_id_ssm(region_name, parameter_path):
 )
 AML2_CPU_ARM64_US_EAST_1 = get_ami_id_boto3(
     region_name="us-east-1",
-    ami_name_pattern="Deep Learning Base AMI (Amazon Linux 2) Version ??.?",
+    ami_name_pattern="Deep Learning ARM64 Base OSS Nvidia Driver GPU AMI (Amazon Linux 2) ????????",
 )
 PT_GPU_PY3_BENCHMARK_IMAGENET_AMI_US_EAST_1 = "ami-0673bb31cc62485dd"
 PT_GPU_PY3_BENCHMARK_IMAGENET_AMI_US_WEST_2 = "ami-02d9a47bc61a31d43"

Original file line number	Diff line number	Diff line change
`@@ -142,7 +142,7 @@ def get_ami_id_ssm(region_name, parameter_path):`
`142`	`142`	`)`
`143`	`143`	`AML2_CPU_ARM64_US_EAST_1 = get_ami_id_boto3(`
`144`	`144`	`region_name="us-east-1",`
`145`		`- ami_name_pattern="Deep Learning Base AMI (Amazon Linux 2) Version ??.?",`
	`145`	`+ ami_name_pattern="Deep Learning ARM64 Base OSS Nvidia Driver GPU AMI (Amazon Linux 2) ????????",`
`146`	`146`	`)`
`147`	`147`	`PT_GPU_PY3_BENCHMARK_IMAGENET_AMI_US_EAST_1 = "ami-0673bb31cc62485dd"`
`148`	`148`	`PT_GPU_PY3_BENCHMARK_IMAGENET_AMI_US_WEST_2 = "ami-02d9a47bc61a31d43"`