Skip to content

Commit 7bdb058

Browse files
committed
refactor: make Dag._node_to_operation private
1 parent 6cb062f commit 7bdb058

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tdp/core/dag.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def graph(self) -> nx.DiGraph:
6060
"""DAG graph."""
6161
return self._graph
6262

63-
def node_to_operation(
63+
def _node_to_operation(
6464
self, node: str, restart: bool = False, stop: bool = False
6565
) -> Operation:
6666
# ? Restart operations are now stored in collections.operations they can be
@@ -152,7 +152,7 @@ def topological_sort(
152152
"""
153153
return list(
154154
map(
155-
lambda node: self.node_to_operation(node, restart=restart, stop=stop),
155+
lambda node: self._node_to_operation(node, restart=restart, stop=stop),
156156
self.topological_sort_key(nodes),
157157
)
158158
)
@@ -234,7 +234,7 @@ def get_operation_descendants(
234234
nodes_filtered = filter(lambda node: node not in nodes, nodes_set)
235235
return list(
236236
map(
237-
lambda node: self.node_to_operation(node, restart=restart, stop=stop),
237+
lambda node: self._node_to_operation(node, restart=restart, stop=stop),
238238
nodes_filtered,
239239
)
240240
)

0 commit comments

Comments
 (0)