code: def setUp(self):
      super().setUp()
      self.model_stub_patcher = mock.patch(
          "tensorflow_serving.apis.model_service_pb2_grpc.ModelServiceStub"
      )  # pylint: disable=line-too-long
      self.model_stub_cls = self.model_stub_patcher.start()
      self.model_stub = self.model_stub_cls.return_value
      self.prediction_stub_patcher = mock.patch(
          "tensorflow_serving.apis.prediction_service_pb2_grpc.PredictionServiceStub"
      )  # pylint: disable=line-too-long
      self.prediction_stub_cls = self.prediction_stub_patcher.start()
      self.prediction_stub = self.prediction_stub_cls.return_value
  
imports: from typing import Any, Dict
  from unittest import mock
  import grpc
  import tensorflow
  from tfx.components.infra_validator import error_types
  from tfx.components.infra_validator import types
  from tfx.components.infra_validator.model_server_clients import tensorflow_serving_client
  from google.protobuf import json_format
  from tensorflow_serving.apis import classification_pb2
  from tensorflow_serving.apis import get_model_status_pb2
  from tensorflow_serving.apis import regression_pb2
label_desc: Use image node_id 1 for calling the TensorflowServingClientTest obj's underlying member method code with example usage: obj.setUp() without return types