fix test

Author: pagezyhf

tests/unit/sagemaker/image_uris/test_huggingface_llm.py

@@ -71,8 +71,8 @@
         "0.0.25": "2.1.2-optimum0.0.25-neuronx-py310-ubuntu22.04",
         "0.0.27": "2.1.2-optimum0.0.27-neuronx-py310-ubuntu22.04",
         "0.0.28": "2.1.2-optimum0.0.28-neuronx-py310-ubuntu22.04",
-        "0.2.0": "2.1.2-optimum0.2.0-neuronx-py310-ubuntu22.04",
-        "0.3.0": "2.7.0-optimum3.3.6-neuronx-py310-ubuntu22.04"
+        "0.2.0": "2.5.1-optimum3.3.4-neuronx-py310-ubuntu22.04",
+        "0.3.0": "2.7.0-optimum3.3.6-neuronx-py310-ubuntu22.04",
     },
 }