message: Fix test failure in tf2.0 that std server cannot be started.
diff: @@ -290,8 +290,13 @@ class MultiWorkerTestBase(test.TestCase):
  return config
  - def _run_client(self, client_fn, task_type, task_id, num_gpus, *args,
  - **kwargs):
  + def _run_client(self, client_fn, task_type, task_id, num_gpus, eager_mode,
  + *args, **kwargs):
  + if eager_mode:
  + with context.eager_mode():
  + result = client_fn(task_type, task_id, num_gpus, *args, **kwargs)
  + else:
  + with context.graph_mode():
  result = client_fn(task_type, task_id, num_gpus, *args, **kwargs)
  if np.all(result):
  with self._lock:
  @@ -314,7 +319,8 @@ class MultiWorkerTestBase(test.TestCase):
  for task_id in range(len(cluster_spec.get(task_type, []))):
  t = threading.Thread(
  target=self._run_client,
  - args=(client_fn, task_type, task_id, num_gpus) + args,
  + args=(client_fn, task_type, task_id, num_gpus,
  + context.executing_eagerly()) + args,
  kwargs=kwargs)
  t.start()
  threads.append(t)
  