Fixed bug when exporting to graphviz dot format as undirected graph (#…

…201)
caesar0301 · Mar 27, 2023 · 226dd03 · 226dd03
1 parent f11c146
commit 226dd03
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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