X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=agtree2dot.git;a=blobdiff_plain;f=agtree2dot.py;h=2dcc1d705752bf185248aa228b045a10a88c2c4f;hp=4eef05a4fca1cd967af795b817a3827fd0cdd7d8;hb=9d6a57fb731d72261e4e0a334595cc5745765e2a;hpb=75c4f522cda71ef830148ec8d881157322be9502

diff --git a/agtree2dot.py b/agtree2dot.py
index 4eef05a..2dcc1d7 100755
--- a/agtree2dot.py
+++ b/agtree2dot.py
@@ -92,21 +92,15 @@ def fill_graph_lists(u, node_labels, node_list, link_list):
             add_link(node_list, link_list, u, 0, u.variable, 0)
 
         if hasattr(u, 'next_functions'):
-            i = 0
-            for v, j in u.next_functions:
+            for i, (v, j) in enumerate(u.next_functions):
                 fill_graph_lists(v, node_labels, node_list, link_list)
                 add_link(node_list, link_list, u, i, v, j)
-                i += 1
 
 ######################################################################
 
 def print_dot(node_list, link_list, out):
     out.write('digraph{\n')
 
-    out.write('  graph [fontname = "helvetica"];\n')
-    out.write('  node [fontname = "helvetica"];\n')
-    out.write('  edge [fontname = "helvetica"];\n')
-
     for n in node_list:
         node = node_list[n]