code: def setUp(self):
      super(EagerClusterReplicateTest, self).setUp()
  
      # TODO(b/171412104): Move create server to __init__ once tfrt support it.
      self._cached_server1 = server_lib.Server.create_local_server()
      self._cached_server2 = server_lib.Server.create_local_server()
      self._cached_server1_target = self._cached_server1.target[
          len("grpc://") :
      ]
      self._cached_server2_target = self._cached_server2.target[
          len("grpc://") :
      ]
  
      # Start the local server.
      local_port = pywrap_tfe.TF_PickUnusedPortOrDie()
      context.set_server_def(
          server_def=_get_server_def(
              self._job_name,
              local_server_port=local_port,
              remote_server_addresses=[
                  self._cached_server1_target,
                  self._cached_server2_target,
              ],
              task_index=0,
          )
      )
  
imports: from absl.testing import parameterized
  from tensorflow.core.protobuf import cluster_pb2
  from tensorflow.core.protobuf import config_pb2
  from tensorflow.core.protobuf import tensorflow_server_pb2
  from tensorflow.python import pywrap_tfe
  from tensorflow.python.client import session
  from tensorflow.python.data.experimental.ops import distribute
  from tensorflow.python.data.kernel_tests import test_base
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.data.ops import options
  from tensorflow.python.eager import context
  from tensorflow.python.framework import combinations
  from tensorflow.python.framework import config
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops import variable_scope
  from tensorflow.python.platform import test
  from tensorflow.python.training import server_lib
label_desc: Use image node_id 2 for calling the EagerClusterReplicateTest obj's underlying member method code with example usage: obj.setUp() without return types