code: def prepare_triton_builder_for_model(
      self, triton_builder: Triton
  ) -> Triton:
      triton_builder.model = MOCK_PT_MODEL
      triton_builder.image_uri = TRITON_IMAGE
      triton_builder.mode = Mode.LOCAL_CONTAINER
      triton_builder.schema_builder = pt_schema_builder
      triton_builder.model_path = MODEL_PATH
      triton_builder.s3_upload_path = S3_UPLOAD_PATH
      triton_builder.serve_settings = MOCK_SERVE_SETTINGS
      triton_builder.env_vars = ENV_VAR
      triton_builder.sagemaker_session = MOCK_SESSION
      triton_builder.modes = MOCK_MODES
      triton_builder._model_builder_deploy_wrapper = MOCK_DEPLOY_WRAPPER
      triton_builder._model_builder_register_wrapper = (
          MOCK_RESIGTER_WRAPPER
      )
      triton_builder.inference_spec = None
  
      mock_export = Mock()
      triton_builder._export_pytorch_to_onnx = mock_export
      triton_builder._export_tf_to_onnx = mock_export
  
      return triton_builder
  
file_path: sagemaker-python-sdk/tests/unit/sagemaker/serve/model_server/triton/test_triton_builder.py
imports: from __future__ import absolute_import
  from unittest import TestCase
  from unittest.mock import Mock, patch
  from sagemaker.serve.model_server.triton.triton_builder import Triton
  from sagemaker.serve.builder.schema_builder import SchemaBuilder
  from sagemaker.serve.mode.function_pointers import Mode
  import torch
label_desc: Use image node_id 1 for calling the TritonBuilderTests obj's underlying member method code with example usage: obj.prepare_triton_builder_for_model(triton_builder) and returns: triton_builder