summaryrefslogtreecommitdiff
path: root/taskflow/patterns/graph_flow.py
diff options
context:
space:
mode:
authorMichael Johnson <johnsomor@gmail.com>2019-10-18 18:02:58 -0700
committerMichael Johnson <johnsomor@gmail.com>2019-10-18 18:11:44 -0700
commitdc6495cfa1c8e1dc95bad554a55f0b4e8e360abe (patch)
treec198a067723a292c9ed9991a317d14d2278911d9 /taskflow/patterns/graph_flow.py
parent9422017336af97c799c611930ddb9dfd0fb0e47b (diff)
downloadtaskflow-dc6495cfa1c8e1dc95bad554a55f0b4e8e360abe.tar.gz
Update TaskFlow for networkx 2.x
The networkx 2.x series has been out for two years now and supports python 3.6 and greater[1]. This patch updates TaskFlow to require a minimum of networkx 2.1. It also updates the code to support recent deprecation expiration introduced in the 2.4 release. [1] https://networkx.github.io/documentation/stable/news.html Change-Id: Ife31d353ba80824ebc63c8b21ee90943badc8da3
Diffstat (limited to 'taskflow/patterns/graph_flow.py')
-rw-r--r--taskflow/patterns/graph_flow.py11
1 files changed, 6 insertions, 5 deletions
diff --git a/taskflow/patterns/graph_flow.py b/taskflow/patterns/graph_flow.py
index ea45f85..52c8178 100644
--- a/taskflow/patterns/graph_flow.py
+++ b/taskflow/patterns/graph_flow.py
@@ -224,7 +224,7 @@ class Flow(flow.Flow):
retry_provides.add(value)
provided[value].append(self._retry)
- for node in self._graph.nodes_iter():
+ for node in self._graph.nodes:
for value in self._unsatisfied_requires(node, self._graph,
retry_provides):
required[value].append(node)
@@ -292,12 +292,12 @@ class Flow(flow.Flow):
yield n
def iter_links(self):
- return self._get_subgraph().edges_iter(data=True)
+ return self._get_subgraph().edges(data=True)
def iter_nodes(self):
g = self._get_subgraph()
for n in g.topological_sort():
- yield n, g.node[n]
+ yield n, g.nodes[n]
@property
def requires(self):
@@ -307,7 +307,7 @@ class Flow(flow.Flow):
requires.update(self._retry.requires)
retry_provides.update(self._retry.provides)
g = self._get_subgraph()
- for node in g.nodes_iter():
+ for node in g.nodes:
requires.update(self._unsatisfied_requires(node, g,
retry_provides))
return frozenset(requires)
@@ -367,6 +367,7 @@ class TargetedFlow(Flow):
return self._graph
nodes = [self._target]
nodes.extend(self._graph.bfs_predecessors_iter(self._target))
- self._subgraph = gr.DiGraph(data=self._graph.subgraph(nodes))
+ self._subgraph = gr.DiGraph(
+ incoming_graph_data=self._graph.subgraph(nodes))
self._subgraph.freeze()
return self._subgraph