Skip to content

Commit 474437c

Browse files
authored
Merge pull request #240 from py-why/fix_duplicate_pydot_nodes
fix duplicate pydot nodes
2 parents 5b3ab2e + 8c54295 commit 474437c

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

causallearn/utils/GraphUtils.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -536,10 +536,9 @@ def to_pydot(G: Graph, edges: List[Edge] | None = None, labels: List[str] | None
536536

537537
pydot_g = pydot.Dot(title, graph_type="digraph", fontsize=18)
538538
pydot_g.obj_dict["attributes"]["dpi"] = dpi
539-
nodes = G.get_nodes()
539+
540540
for i, node in enumerate(nodes):
541541
node_name = labels[i] if labels is not None else node.get_name()
542-
pydot_g.add_node(pydot.Node(i, label=node.get_name()))
543542
if node.get_node_type() == NodeType.LATENT:
544543
pydot_g.add_node(pydot.Node(i, label=node_name, shape='square'))
545544
else:

0 commit comments

Comments
 (0)