lint

intelligent-machine-learning · Oct 12, 2024 · a3b5b17 · a3b5b17
1 parent 818c05a
commit a3b5b17
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/dlrover/python/master/node/dist_job_manager.py b/dlrover/python/master/node/dist_job_manager.py
@@ -1138,9 +1138,11 @@ def update_node_required_info_callback(self):
 
  def update_succeeded_node(self, node_id, node_type):
  with self._lock:
- if node_type in self._job_nodes:
- if node_id in self._job_nodes[node_type]:
- self._job_nodes[node_type][node_id].set_as_succeeded()
+ if (
+ node_type in self._job_nodes
+ and node_id in self._job_nodes[node_type]
+ ):
+ self._job_nodes[node_type][node_id].set_as_succeeded()
 
 
 def create_job_manager(args: JobArgs, speed_monitor) -> DistributedJobManager:

diff --git a/dlrover/python/master/node/job_manager.py b/dlrover/python/master/node/job_manager.py
@@ -233,6 +233,8 @@ def get_elastic_run_configs(self) -> Dict[str, str]:
  return self._training_node_config.get_elastic_run_configs()
 
  def update_succeeded_node(self, node_id, node_type):
- if node_type in self._job_nodes:
- if node_id in self._job_nodes[node_type]:
- self._job_nodes[node_type][node_id].set_as_succeeded()
+ if (
+ node_type in self._job_nodes
+ and node_id in self._job_nodes[node_type]
+ ):
+ self._job_nodes[node_type][node_id].set_as_succeeded()