code: def test_smp_v2(load_config):
      VERSIONS = load_config["training"]["versions"]
      PROCESSORS = load_config["training"]["processors"]
      distribution = {
          "torch_distributed": {"enabled": True},
          "smdistributed": {"modelparallel": {"enabled": True}},
      }
      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:
                  for region in ACCOUNTS.keys():
                      for instance_type in CONTAINER_VERSIONS.keys():
                          uri = image_uris.get_training_image_uri(
                              region,
                              framework="pytorch",
                              framework_version=version,
                              py_version=py_version,
                              distribution=distribution,
                              instance_type=instance_type,
                          )
                          expected = expected_uris.framework_uri(
                              repo="smdistributed-modelparallel",
                              fw_version=version,
                              py_version=f"{py_version}-{CONTAINER_VERSIONS[instance_type]}",
                              processor=processor,
                              region=region,
                              account=ACCOUNTS[region],
                          )
                          assert expected == uri
  
imports: from __future__ import absolute_import
  import pytest
  from sagemaker import image_uris
  from tests.unit.sagemaker.image_uris import expected_uris
label_desc: Use image node_id 1 for calling a global function with example usage: test_smp_v2(load_config) without return types