def setUpClass(cls):
   cls._workers, cls._ps = multi_worker_test_base.create_in_process_cluster(
       num_workers=3, num_ps=2)
   cls._cluster_spec = {
       run_config.TaskType.WORKER: [
           'fake_worker_0', 'fake_worker_1', 'fake_worker_2'
       ],
       run_config.TaskType.PS: ['fake_ps_0', 'fake_ps_1']
   }

 def setUpClass(cls):
   """Create a local cluster with 2 workers."""
   cls._workers, cls._ps = multi_worker_test_base.create_in_process_cluster(
       num_workers=3, num_ps=0)
   cls._cluster_spec = {
       run_config.TaskType.WORKER: [
           "fake_worker_0", "fake_worker_1", "fake_worker_2"
       ]
   }

 def setUpClass(cls):
   cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
       num_workers=3, num_ps=2, has_chief=True)
   cls._default_target = 'grpc://' + cls._cluster_spec[CHIEF][0]

 def setUpClass(cls):
   """Create a local cluster with 3 workers and 1 chief."""
   cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
       num_workers=3, num_ps=0, has_chief=True)

 def setUpClass(cls):
   """Create a local cluster with 2 workers."""
   super(DistributeCoordinatorIntegrationTest, cls).setUpClass()
   cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
       num_workers=3, num_ps=2, has_eval=True)

 def setUpClass(cls):
   """Create a local cluster with 2 workers."""
   cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
       num_workers=3, num_ps=0)

 def setUpClass(cls):
   """Create a local cluster with 2 workers and 1 chief."""
   cls._cluster_spec = multi_worker_test_base.create_in_process_cluster(
       num_workers=2, num_ps=0, has_chief=True)
   cls._default_target = "grpc://" + cls._cluster_spec["chief"][0]

 def setUpClass(cls):
   """Create a local cluster with 2 workers."""
   super(KerasMultiWorkerTestStandaloneClient, cls).setUpClass()
   cls._cluster_spec = test_base.create_in_process_cluster(
       num_workers=2, num_ps=1, has_eval=False)