fix: update neuronx huggingface tests (#5314)

* fix: update neuronx huggingface tests

* black formatter

* Add inference_ami_version=al2-ami-sagemaker-inference-neuron-2 to all neuronx tests

---------

Co-authored-by: Sirut Buasai <73297481+sirutBuasai@users.noreply.github.com>

Author: ahsan-z-khan

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 "}

test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_neuronx_encoder_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": "Hamilton is considered to be the best musical of human history."}

test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_neuronx_no_context.py

@@ -127,6 +127,7 @@ def _test_sentence_transformers(
             initial_instance_count=1,
             instance_type=instance_type,
             endpoint_name=endpoint_name,
+            inference_ami_version="al2-ami-sagemaker-inference-neuron-2",
         )
 
         predictor.serializer = IdentitySerializer(content_type="application/json")

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"}