code: def test_build_for_triton_tf(
      self, mock_detect_fw, mock_get_gpus, mock_path, mock_model
  ):
      triton_builder = self.prepare_triton_builder_for_model(
          triton_builder=Triton()
      )
  
      mock_model_path = Mock()
      mock_path.return_value = mock_model_path
      mock_model_path.exists.return_value = True
      mock_model_path.is_dir.return_value = True
      mock_model_path.joinpath.return_value = mock_model_path
  
      mock_detect_fw.return_value = ("tensorflow", "2.0.1")
  
      triton_builder.model = MOCK_TF_MODEL
      triton_builder._build_for_triton()
  
      triton_builder._export_tf_to_onnx.assert_called_once_with(
          export_path=mock_model_path,
          model=MOCK_TF_MODEL,
          schema_builder=pt_schema_builder,
      )
  
      mock_model.assert_called_with(
          image_uri=TRITON_IMAGE,
          model_data=S3_UPLOAD_PATH,
          role=ROLE_ARN,
          env=ENV_VAR,
          sagemaker_session=MOCK_SESSION,
          predictor_cls=triton_builder._get_triton_predictor,
      )
  
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 3 for calling the TritonBuilderTests obj's underlying member method code with example usage: obj.test_build_for_triton_tf(mock_detect_fw, mock_get_gpus, mock_path, mock_model) without return types