OCD cosmetics.

[agtree2dot.git] / agtree2dot.py

diff --git a/agtree2dot.py b/agtree2dot.py
index 24d5f85..2dcc1d7 100755 (executable)
--- a/agtree2dot.py
+++ b/agtree2dot.py
@@ -92,11 +92,9 @@ 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
 
 ######################################################################