printing variables

Author: adtian2

tests/unit/sagemaker/image_uris/test_smp_v2.py

@@ -27,6 +27,10 @@ def test_smp_v2(load_config):
         "torch_distributed": {"enabled": True},
         "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"]
@@ -42,6 +46,8 @@ 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",