def testWorkerNotInCluster(self): cluster_spec = { "chief": ["127.0.0.1:1234"], "ps": ["127.0.0.1:1926", "127.0.0.1:3141"] } multi_worker_util._validate_cluster_spec(cluster_spec, "evaluator", 0) with self.assertRaisesRegex( ValueError, "`task_type` 'worker' not found in cluster_spec."): multi_worker_util._validate_cluster_spec(cluster_spec, "worker", 0)
def _verify_args_and_config(self, cluster_resolver): if not cluster_resolver.cluster_spec(): raise ValueError("Cluster spec must be non-empty in " "`tf.distribute.cluster_resolver.ClusterResolver`.") cluster_spec = cluster_resolver.cluster_spec() # The following checks if the task types are allowed (chief, ps, worker). multi_worker_util._validate_cluster_spec( # pylint: disable=protected-access cluster_spec, cluster_resolver.task_type, cluster_resolver.task_id) if multi_worker_util.task_count(cluster_spec, "ps") < 1: raise ValueError("There must be at least one ps.") if multi_worker_util.task_count(cluster_spec, "worker") < 1: raise ValueError("There must be at least one worker.")
def testEvaluatorNotInCluster(self): cluster_spec = { "chief": ["127.0.0.1:1234"], "worker": ["127.0.0.1:8964", "127.0.0.1:2333"], "ps": ["127.0.0.1:1926", "127.0.0.1:3141"] } multi_worker_util._validate_cluster_spec(cluster_spec, "chief", 0) multi_worker_util._validate_cluster_spec(cluster_spec, "worker", 0) multi_worker_util._validate_cluster_spec(cluster_spec, "ps", 0) multi_worker_util._validate_cluster_spec(cluster_spec, "evaluator", 0)