code: def _expected_trainium_framework_uri(
      framework,
      version,
      account,
      region="us-west-2",
      inference_tool="neuron",
  ):
      return expected_uris.neuron_framework_uri(
          "{}-neuron".format(framework),
          fw_version=version,
          py_version="py38",
          account=account,
          region=region,
          inference_tool=inference_tool,
      )
  
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 1 for calling a global function with example usage: _expected_trainium_framework_uri(framework, version, account, region, inference_tool) and returns: expected_uris