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

version

Author: adtian2

src/sagemaker/image_uri_config/pytorch-smp.json

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

tests/unit/sagemaker/image_uris/test_smp_v2.py

@@ -28,16 +28,13 @@ def test_smp_v2(load_config):
         "smdistributed": {"modelparallel": {"enabled": True}},
     }
 
-    print("load_config", load_config)
-    print("VERSIONS", VERSIONS)
-
     for processor in PROCESSORS:
         for version in VERSIONS:
             ACCOUNTS = load_config["training"]["versions"][version]["registries"]
             PY_VERSIONS = load_config["training"]["versions"][version]["py_versions"]
             for py_version in PY_VERSIONS:
-                # py311 is only for smp version 2.4.0
-                if py_version == "py311" and "2.4" not in version:
+                # py311 is only for PT 2.3.1 and SMP 2.4.0
+                if py_version == "py311" and "2.3" not in version:
                     continue
 
                 for region in ACCOUNTS.keys():
@@ -46,8 +43,6 @@ def test_smp_v2(load_config):
                         if "2.1" in version or "2.2" in version or "2.3" in version:
                             cuda_vers = "cu121"
 
-                        print("HERE", version, py_version)
-
                         uri = image_uris.get_training_image_uri(
                             region,
                             framework="pytorch",