Commit df3b1c36 authored by VIGNET Pierre's avatar VIGNET Pierre
Browse files

Fix colors for matplotlib graph export

parent b88a1197
......@@ -652,8 +652,9 @@ def draw_graph(output_dir, solution, solution_index, G,
node_color=colors, node_size=1000, alpha=0.5, ax=ax)
# Draw edges involved in transitions with conditions
edges_colors = [edge[2]['color'] for edge in edges_in_cond]
nx.draw_networkx_edges(G, pos, edgelist=edges_in_cond,
edge_color='b', width=2, alpha=0.5)
edge_color=edges_colors, width=2, alpha=0.5)
# Draw labels for normal transitions (move pos to the end of the arrow)
# ex: [('Ax', 'n1', {u'condition': u'', u'label': u'h00[]'}),]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment