formatting if

exo-explore · Jul 20, 2024 · 9785e25 · 9785e25
1 parent 7708b47
commit 9785e25
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/exo/orchestration/standard_node.py b/exo/orchestration/standard_node.py
@@ -40,8 +40,7 @@ def on_node_status(self, request_id, opaque_status):
                 elif status_data.get("status", "").startswith("end_"):
                     if status_data.get("node_id") == self.current_topology.active_node_id:
                         self.current_topology.active_node_id = None
-            if self.topology_viz:
-              self.topology_viz.update_visualization(self.current_topology, self.partitioning_strategy.partition(self.current_topology))
+            if self.topology_viz: self.topology_viz.update_visualization(self.current_topology, self.partitioning_strategy.partition(self.current_topology))
         except json.JSONDecodeError:
             pass
 
@@ -243,8 +242,7 @@ async def collect_topology(self, visited: set[str] = set(), max_depth: int = 4)
 
         next_topology.active_node_id = self.topology.active_node_id # this is not so clean.
         self.topology = next_topology
-        if self.topology_viz:
-            self.topology_viz.update_visualization(self.current_topology, self.partitioning_strategy.partition(self.current_topology))
+        if self.topology_viz: self.topology_viz.update_visualization(self.current_topology, self.partitioning_strategy.partition(self.current_topology))
         return next_topology
 
     # TODO: unify this and collect_topology as global actions