code: def test_attach_custom_image(sagemaker_session):
      training_image = "pytorch:latest"
      returned_job_description = {
          "AlgorithmSpecification": {
              "TrainingInputMode": "File",
              "TrainingImage": training_image,
          },
          "HyperParameters": {
              "sagemaker_submit_directory": '"s3://some/sourcedir.tar.gz"',
              "sagemaker_program": '"iris-dnn-classifier.py"',
              "sagemaker_s3_uri_training": '"sagemaker-3/integ-test-data/tf_iris"',
              "sagemaker_container_log_level": '"logging.INFO"',
              "sagemaker_job_name": '"neo"',
              "training_steps": "100",
              "sagemaker_region": '"us-east-1"',
          },
          "RoleArn": "arn:aws:iam::366:role/SageMakerRole",
          "ResourceConfig": {
              "VolumeSizeInGB": 30,
              "InstanceCount": 1,
              "InstanceType": "ml.c4.xlarge",
          },
          "StoppingCondition": {"MaxRuntimeInSeconds": 24 * 60 * 60},
          "TrainingJobName": "neo",
          "TrainingJobStatus": "Completed",
          "TrainingJobArn": "arn:aws:sagemaker:us-west-2:336:training-job/neo",
          "OutputDataConfig": {
              "KmsKeyId": "",
              "S3OutputPath": "s3://place/output/neo",
          },
          "TrainingJobOutput": {
              "S3TrainingJobOutput": "s3://here/output.tar.gz"
          },
      }
      sagemaker_session.sagemaker_client.describe_training_job = Mock(
          name="describe_training_job",
          return_value=returned_job_description,
      )
  
      estimator = HuggingFace.attach(
          training_job_name="neo", sagemaker_session=sagemaker_session
      )
      assert estimator.latest_training_job.job_name == "neo"
      assert estimator.image_uri == training_image
  
file_path: sagemaker-python-sdk/tests/unit/sagemaker/huggingface/test_estimator.py
imports: from __future__ import absolute_import
  import logging
  import json
  import os
  import pytest
  from mock import MagicMock, Mock, patch
  from sagemaker.huggingface import HuggingFace, HuggingFaceModel
  from sagemaker.session_settings import SessionSettings
  from .huggingface_utils import get_full_gpu_image_uri, GPU_INSTANCE_TYPE, REGION
label_desc: Use image node_id 9 for calling a global function with example usage: test_attach_custom_image(sagemaker_session) without return types