code: def mock_sagemaker_client():
      mock_sagemaker_client = MagicMock()
      mock_sagemaker_client.create_training_job = (
          mock_create_training_job
      )
      mock_sagemaker_client.create_training_job.return_value = {
          "TrainingJobArn": "arn:aws:sagemaker:us-east-1:123456789012:TrainingJob/test-job-1"
      }
      mock_sagemaker_client.stop_training_job.return_value = {
          "TrainingJobArn": "arn:aws:sagemaker:us-east-1:123456789012:TrainingJob/test-job-1"
      }
      mock_sagemaker_client.describe_training_job.return_value = {
          "TrainingJobStatus": "Completed",
          "TrainingJobName": "test-job-1",
      }
      return mock_sagemaker_client
  
file_path: wandb/tests/pytest_tests/unit_tests_old/test_launch/test_launch_aws.py
imports: import json
  import sys
  from unittest.mock import MagicMock
  import pytest
  import wandb
  import wandb.sdk.launch._launch
  import wandb.sdk.launch._project_spec
  from wandb.sdk.launch.environment.aws_environment import AwsEnvironment
  from wandb.sdk.launch.errors import LaunchError
  from wandb.sdk.launch.runner.sagemaker_runner import SagemakerSubmittedRun
  from tests.pytest_tests.unit_tests_old.utils import fixture_open
  from .test_launch import mocked_fetchable_git_repo
label_desc: Use image node_id 4 for calling a global function with example usage: mock_sagemaker_client() and returns: mock_sagemaker_client