fix: update neuronx huggingface tests

ahsan-z-khan · ahsan-z-khan · commit 43d09ef9d9d6 · 2025-09-29T09:06:52.000-07:00
diff --git a/test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_neuronx_decoder_hosting.py b/test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_neuronx_decoder_hosting.py
@@ -145,6 +145,7 @@ def _test_pt_neuronx(
             initial_instance_count=1,
             instance_type=instance_type,
             endpoint_name=endpoint_name,
+            inference_ami_version="al2-ami-sagemaker-inference-neuron-2",
         )
 
         data = {"inputs": "I really wish "}
diff --git a/test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_neuronx_sdxl_hosting.py b/test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_neuronx_sdxl_hosting.py
@@ -146,6 +146,7 @@ def _test_pt_neuronx(
             initial_instance_count=1,
             instance_type=instance_type,
             endpoint_name=endpoint_name,
+            inference_ami_version="al2-ami-sagemaker-inference-neuron-2"
         )
 
         data = {"inputs": "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"}

Original file line number	Diff line number	Diff line change
`@@ -145,6 +145,7 @@ def _test_pt_neuronx(`
`145`	`145`	`initial_instance_count=1,`
`146`	`146`	`instance_type=instance_type,`
`147`	`147`	`endpoint_name=endpoint_name,`
	`148`	`+ inference_ami_version="al2-ami-sagemaker-inference-neuron-2",`
`148`	`149`	`)`
`149`	`150`
`150`	`151`	`data = {"inputs": "I really wish "}`
Original file line number	Diff line number	Diff line change
`@@ -146,6 +146,7 @@ def _test_pt_neuronx(`
`146`	`146`	`initial_instance_count=1,`
`147`	`147`	`instance_type=instance_type,`
`148`	`148`	`endpoint_name=endpoint_name,`
	`149`	`+ inference_ami_version="al2-ami-sagemaker-inference-neuron-2"`
`149`	`150`	`)`
`150`	`151`
`151`	`152`	`data = {"inputs": "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"}`