From 068b4af4baacbf2be13fcbaa702361728517c83b Mon Sep 17 00:00:00 2001 From: Paul Farault Date: Wed, 20 Nov 2024 12:10:43 +0100 Subject: [PATCH] refactor: make Dag._node_to_operation private --- tdp/core/dag.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tdp/core/dag.py b/tdp/core/dag.py index 84584a00..2398cee6 100644 --- a/tdp/core/dag.py +++ b/tdp/core/dag.py @@ -60,7 +60,7 @@ def graph(self) -> nx.DiGraph: """DAG graph.""" return self._graph - def node_to_operation( + def _node_to_operation( self, node: str, restart: bool = False, stop: bool = False ) -> Operation: # ? Restart operations are now stored in collections.operations they can be @@ -152,7 +152,7 @@ def topological_sort( """ return list( map( - lambda node: self.node_to_operation(node, restart=restart, stop=stop), + lambda node: self._node_to_operation(node, restart=restart, stop=stop), self.topological_sort_key(nodes), ) ) @@ -234,7 +234,7 @@ def get_operation_descendants( nodes_filtered = filter(lambda node: node not in nodes, nodes_set) return list( map( - lambda node: self.node_to_operation(node, restart=restart, stop=stop), + lambda node: self._node_to_operation(node, restart=restart, stop=stop), nodes_filtered, ) )