version alias conficts resolved

Author: adtian2

src/sagemaker/image_uri_config/pytorch-smp.json

@@ -8,7 +8,6 @@
             "2.1": "2.1.2",
             "2.2": "2.3.1",
             "2.2.0": "2.3.1",
-            "2.3": "2.4.0",
             "2.3.1": "2.4.0"
         },
         "versions": {

tests/unit/sagemaker/image_uris/test_smp_v2.py

@@ -39,7 +39,7 @@ def test_smp_v2(load_config):
                         if "2.1" in version or "2.2" in version or "2.3" in version:
                             cuda_vers = "cu121"
 
-                        if "2.3" in version:
+                        if "2.3.1" == version:
                             py_version = "py311"
 
                         print(version, py_version)