PY3 fixed d3viz in Python 3

theano/d3viz/d3viz.py

@@ -7,6 +7,7 @@ from __future__ import absolute_import, print_function, division
 import os
 import shutil
 import re
+import six
 from six import iteritems
 
 from theano.d3viz.formatting import PyDotFormatter
@@ -78,7 +79,10 @@ def d3viz(fct, outfile, copy_deps=True, *args, **kwargs):
     # Create DOT graph
     formatter = PyDotFormatter(*args, **kwargs)
     graph = formatter(fct)
-    dot_graph = escape_quotes(str(graph.create_dot())).replace('\n', '').replace('\r', '')
+    dot_graph_raw = graph.create_dot()
+    if not six.PY2:
+        dot_graph_raw = dot_graph_raw.decode('utf8')
+    dot_graph = escape_quotes(dot_graph_raw).replace('\n', '').replace('\r', '')
 
     # Create output directory if not existing
     outdir = os.path.dirname(outfile)