From 226dd03c32c20c7a372a92a08217398a89b778cd Mon Sep 17 00:00:00 2001 From: tsoj Date: Mon, 27 Mar 2023 13:26:06 +0200 Subject: [PATCH] Fixed bug when exporting to graphviz dot format as undirected graph (#201) --- treelib/tree.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/treelib/tree.py b/treelib/tree.py index 20109bc..3a64779 100644 --- a/treelib/tree.py +++ b/treelib/tree.py @@ -971,7 +971,8 @@ def to_graphviz(self, filename=None, shape='circle', graph='digraph', for c in self.children(nid): cid = c.identifier - connections.append('"{0}" -> "{1}"'.format(nid, cid)) + edge = '->' if graph == 'digraph' else '--' + connections.append(('"{0}" ' + edge + ' "{1}"').format(nid, cid)) # write nodes and connections to dot format is_plain_file = filename is not None