code: def test_training_image_uri(
      retrieve_image_uri, sagemaker_session, ray_tensorflow_version
  ):
      toolkit = RLToolkit.RAY
      framework = RLFramework.TENSORFLOW
  
      image = "custom-image:latest"
      rl = _rl_estimator(
          sagemaker_session,
          toolkit,
          ray_tensorflow_version,
          framework,
          instance_type=CPU,
          image_uri=image,
      )
      assert image == rl.training_image_uri()
      retrieve_image_uri.assert_not_called()
  
      rl = _rl_estimator(
          sagemaker_session,
          toolkit,
          ray_tensorflow_version,
          framework,
          instance_type=CPU,
      )
      assert retrieve_image_uri.return_value == rl.training_image_uri()
  
      retrieve_image_uri.assert_called_with(
          "ray-tensorflow",
          REGION,
          version=ray_tensorflow_version,
          instance_type=CPU,
      )
  
imports: from __future__ import absolute_import
  import json
  import logging
  import os
  import pytest
  from mock import MagicMock, Mock, patch
  from sagemaker.mxnet import MXNetModel, MXNetPredictor
  from sagemaker.rl import RLEstimator, RLFramework, RLToolkit, TOOLKIT_FRAMEWORK_VERSION_MAP
  from sagemaker.session_settings import SessionSettings
  from sagemaker.tensorflow import TensorFlowModel, TensorFlowPredictor
label_desc: Use image node_id 13 for calling a global function with example usage: test_training_image_uri(retrieve_image_uri, sagemaker_session, ray_tensorflow_version) without return types