code: def test_trainium_pytorch_uris(load_config, scope):
      VERSIONS = load_config[scope]["versions"]
      for version in VERSIONS:
          ACCOUNTS = load_config[scope]["versions"][version][
              "registries"
          ]
          for region in ACCOUNTS.keys():
              uri = image_uris.retrieve(
                  "pytorch",
                  region,
                  instance_type="ml.trn1.xlarge",
                  version=version,
              )
              expected = _expected_trainium_framework_uri(
                  "{}-training".format("pytorch"),
                  version,
                  ACCOUNTS[region],
                  region=region,
                  inference_tool="neuron",
              )
              assert expected == uri
  
imports: from __future__ import absolute_import
  from sagemaker import image_uris
  from tests.unit.sagemaker.image_uris import expected_uris
  import pytest
label_desc: Use image node_id 2 for calling a global function with example usage: test_trainium_pytorch_uris(load_config, scope) without return types