X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=agtree2dot.py;h=24d5f853aec1d9b52734ac3a6c672122108701c6;hb=f06a806854deeb2bb50324045a10b4a593927655;hp=0f787accad9edc1d921d839bde6d81a233d6bfbf;hpb=21558202134d649b29fb700481b15963e71b1d1f;p=agtree2dot.git

diff --git a/agtree2dot.py b/agtree2dot.py
index 0f787ac..24d5f85 100755
--- a/agtree2dot.py
+++ b/agtree2dot.py
@@ -83,7 +83,7 @@ def fill_graph_lists(u, node_labels, node_list, link_list):
                     re.search('<class \'(.*\.|)([a-zA-Z0-9_]*)\'>', str(type(u))).group(2))
         node_list[u] = node
 
-        if isinstance(u, torch.autograd.Variable):
+        if hasattr(u, 'grad_fn'):
             fill_graph_lists(u.grad_fn, node_labels, node_list, link_list)
             add_link(node_list, link_list, u, 0, u.grad_fn, 0)
 
@@ -103,10 +103,6 @@ def fill_graph_lists(u, node_labels, node_list, link_list):
 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]