code: def _setUpTrainingMocks(self):
      self._mock_create = mock.Mock()
      self._mock_api_client.projects().jobs().create = self._mock_create
      self._mock_get = mock.Mock()
      self._mock_api_client.projects().jobs().get.return_value = (
          self._mock_get
      )
      self._mock_get.execute.return_value = {
          "state": "SUCCEEDED",
      }
  
file_path: tfx/tfx/orchestration/kubeflow/v2/components/experimental/ai_platform_training_executor_test.py
imports: import copy
  from unittest import mock
  from googleapiclient import discovery
  import tensorflow
  from tfx.dsl.component.experimental import placeholders
  from tfx.orchestration.kubeflow.v2.components.experimental import ai_platform_training_executor
  from tfx.types import artifact_utils
  from tfx.types import standard_artifacts
  from tfx.utils import json_utils
  from tfx.utils import test_case_utils
label_desc: Use image node_id 2 for calling the AiPlatformTrainingExecutorTest obj's underlying member method code with example usage: obj._setUpTrainingMocks() without return types