Fix flake8 errors

theano/d3viz/formatting.py

@@ -8,18 +8,18 @@ import os
 from functools import reduce
 from six import iteritems, itervalues
 
-pydot_installed = True
-try:
-    import pydot as pd
-except ImportError:
-    pydot_installed = False
-
 import theano
 from theano import gof
 from theano.compile.profilemode import ProfileMode
 from theano.compile import Function
 from theano.compile import builders
 
+pydot_installed = True
+try:
+    import pydot as pd
+except ImportError:
+    pydot_installed = False
+
 
 class PyDotFormatter(object):
     """Create `pydot` graph object from Theano function.

theano/tests/test_flake8.py

@@ -216,7 +216,9 @@ whitelist_flake8 = [
     "gof/tests/test_cc.py",
     "gof/tests/test_compute_test_value.py",
     "gof/sandbox/equilibrium.py",
-    "d3viz/__init__.py"
+    "d3viz/__init__.py",
+    "d3viz/tests/test_d3viz.py",
+    "d3viz/tests/test_formatting.py"
 ]