Use master nodes when the job has no chief nodes. (#513)

* Use master nodes when the job has no chief nodes * Format codes
intelligent-machine-learning · Jul 20, 2023 · de49a26 · de49a26
1 parent 574afb2
commit de49a26
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/dlrover/python/master/node/job_manager.py b/dlrover/python/master/node/job_manager.py
@@ -205,8 +205,12 @@ def _init_training_node_manager(self):
  self._elastic_job.get_node_service_addr,
  self._elastic_job.get_node_name,
  )
+
+ chief_nodes = self._job_nodes.get(NodeType.CHIEF, {})
+ if not chief_nodes:
+ chief_nodes = self._job_nodes.get(NodeType.MASTER, {})
  self._chief_manager = ChiefManager(
- self._job_nodes.get(NodeType.CHIEF, {}),
+ chief_nodes,
  self._job_resource,
  self._relaunch_on_worker_failure,
  self._elastic_job.get_node_service_addr,