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, ) )