use super

zincware · Feb 27, 2024 · a1aa11e · a1aa11e
1 parent 0f6c4b0
commit a1aa11e
Showing 1 changed file with 2 additions and 16 deletions.
diff --git a/znflow/deployment/dask_depl.py b/znflow/deployment/dask_depl.py
@@ -56,22 +56,8 @@ class DaskDeployment(DeploymentBase):
     )
 
     def run(self, nodes: t.Optional[list] = None):
-        if nodes is None:
-            for node_uuid in self.graph.reverse():
-                assert self.graph.immutable_nodes
-                node = self.graph.nodes[node_uuid]["value"]
-                self._run_predecessors(node_uuid)
-                self._run_node(node, node_uuid)
-            self._load_results(nodes)
-
-        else:
-            for node_uuid in self.graph.reverse():
-                assert self.graph.immutable_nodes
-                node = self.graph.nodes[node_uuid]["value"]
-                if node in nodes:
-                    self._run_predecessors(node_uuid)
-                    self._run_node(node, node_uuid)
-            self._load_results(nodes)
+        super().run(nodes)
+        self._load_results(nodes)
 
     def _run_node(self, node, node_uuid):
         predecessors = list(self.graph.predecessors(node.uuid))