SMP PT 2.3 Fix (#4796)

* removing 2.3.0 as actual framework version is 2.3.1

* fixed framework_version from 2.3 to 2.3.1

* fixing unit test for PT 2.3 change

* changing to correct smp version in unit test

* changing logic around for unit test

* printing variables

* figured out bug in unit test where version is referring to pt, not smp
version

* trying logic a different way

* version alias conficts resolved

---------

Co-authored-by: Andrew Tian <[email protected]>

Author: adtian2

src/sagemaker/fw_utils.py

@@ -152,7 +152,6 @@
     "2.1.0",
     "2.1.2",
     "2.2.0",
-    "2.3.0",
     "2.3.1",
 ]
 

src/sagemaker/image_uri_config/pytorch-smp.json

@@ -8,7 +8,7 @@
             "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": {
             "2.0.1": {

tests/unit/sagemaker/image_uris/test_smp_v2.py

@@ -27,6 +27,7 @@ def test_smp_v2(load_config):
         "torch_distributed": {"enabled": True},
         "smdistributed": {"modelparallel": {"enabled": True}},
     }
+
     for processor in PROCESSORS:
         for version in VERSIONS:
             ACCOUNTS = load_config["training"]["versions"][version]["registries"]
@@ -38,6 +39,11 @@ 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.1" == version:
+                            py_version = "py311"
+
+                        print(version, py_version)
+
                         uri = image_uris.get_training_image_uri(
                             region,
                             framework="pytorch",