hotfix for canary failures (#4624)

junpuf · web-flow · commit 532271371f81 · 2025-03-10T11:16:21.000-07:00
diff --git a/test/test_utils/__init__.py b/test/test_utils/__init__.py
@@ -164,10 +164,12 @@ def get_ami_id_ssm(region_name, parameter_path):
 UL20_PT_NEURON_US_WEST_2 = get_ami_id_boto3(
     region_name="us-west-2",
     ami_name_pattern="Deep Learning AMI Neuron PyTorch 1.11.0 (Ubuntu 20.04) ????????",
+    IncludeDeprecated=True,
 )
 UL20_TF_NEURON_US_WEST_2 = get_ami_id_boto3(
     region_name="us-west-2",
     ami_name_pattern="Deep Learning AMI Neuron TensorFlow 2.10.? (Ubuntu 20.04) ????????",
+    IncludeDeprecated=True,
 )
 # Since NEURON TRN1 DLAMI is not released yet use a custom AMI
 NEURON_INF1_AMI_US_WEST_2 = "ami-06a5a60d3801a57b7"

Original file line number	Diff line number	Diff line change
`@@ -164,10 +164,12 @@ def get_ami_id_ssm(region_name, parameter_path):`
`164`	`164`	`UL20_PT_NEURON_US_WEST_2 = get_ami_id_boto3(`
`165`	`165`	`region_name="us-west-2",`
`166`	`166`	`ami_name_pattern="Deep Learning AMI Neuron PyTorch 1.11.0 (Ubuntu 20.04) ????????",`
	`167`	`+ IncludeDeprecated=True,`
`167`	`168`	`)`
`168`	`169`	`UL20_TF_NEURON_US_WEST_2 = get_ami_id_boto3(`
`169`	`170`	`region_name="us-west-2",`
`170`	`171`	`ami_name_pattern="Deep Learning AMI Neuron TensorFlow 2.10.? (Ubuntu 20.04) ????????",`
	`172`	`+ IncludeDeprecated=True,`
`171`	`173`	`)`
`172`	`174`	`# Since NEURON TRN1 DLAMI is not released yet use a custom AMI`
`173`	`175`	`NEURON_INF1_AMI_US_WEST_2 = "ami-06a5a60d3801a57b7"`