code: def image_uri(self) -> str:
      if self.framework == "pytorch":
          return "763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-training:1.7.1-transformers4.6.1-gpu-py36-cu110-ubuntu18.04"
      else:
          return "763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-tensorflow-training:2.4.1-transformers4.6.1-gpu-py37-cu110-ubuntu18.04"
  
file_path: H2O/h2o_flexgen/benchmark/third_party/transformers/tests/sagemaker/conftest.py
imports: import os
  import pytest
  from attr import dataclass
label_desc: Use image node_id 4 for calling the SageMakerTestEnvironment obj's underlying member method code with example usage: obj.image_uri() and returns: str, str