Skip to content
Snippets Groups Projects
Commit d63dcc65 authored by Tomáš Orviský's avatar Tomáš Orviský
Browse files

Layout uses key instad of name of object

parent 97545e63
Branches
No related merge requests found
......@@ -18,9 +18,9 @@ class DiagramLayout(ABC):
def get_graph(self) -> nx.DiGraph:
g = nx.DiGraph()
for name, o in self.objects.items():
g.add_node(o.name)
g.add_node(o.get_key())
for e in o.edges:
g.add_edge(e.target.get_key(), o.name)
g.add_edge(e.target.get_key(), o.get_key())
return g
......
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