Apply pyupgrade to theano.d3viz

theano/d3viz/d3viz.py

@@ -8,8 +8,6 @@ import json
 import os
 import shutil
 
-import six
-
 from theano.d3viz.formatting import PyDotFormatter
 
 
@@ -80,8 +78,7 @@ def d3viz(fct, outfile, copy_deps=True, *args, **kwargs):
     formatter = PyDotFormatter(*args, **kwargs)
     graph = formatter(fct)
     dot_graph = graph.create_dot()
-    if not six.PY2:
-        dot_graph = dot_graph.decode("utf8")
+    dot_graph = dot_graph.decode("utf8")
 
     # Create output directory if not existing
     outdir = os.path.dirname(outfile)

theano/d3viz/formatting.py

@@ -19,7 +19,7 @@ except ImportError:
     pass
 
 
-class PyDotFormatter(object):
+class PyDotFormatter:
     """Create `pydot` graph object from Theano function.
 
     Parameters