pep8

theano/gof/tests/test_vm.py

@@ -2,17 +2,12 @@ import gc
 import sys
 import time
 import unittest
-try:
-    import line_profiler
-except ImportError:
-    pass
 
 from nose.plugins.skip import SkipTest
 import numpy
 
 from theano import function
 from theano.gof import vm
-from theano.gof import link
 from theano.gof import OpWiseCLinker
 from theano.compile import Mode
 
@@ -36,14 +31,15 @@ class TestCallbacks(unittest.TestCase):
     def test_callback(self):
         a, b, c = tensor.scalars('abc')
         f = function([a, b, c], (a + b) + c,
-                mode=Mode(
-                    optimizer=None,
-                    linker=vm.VM_Linker(callback=self.callback)))
+                     mode=Mode(
+                         optimizer=None,
+                         linker=vm.VM_Linker(callback=self.callback)))
 
         f(1, 2, 3)
         assert sum(self.n_callbacks.values()) == len(f.maker.fgraph.toposort())
         f(1, 2, 3)
-        assert sum(self.n_callbacks.values()) == len(f.maker.fgraph.toposort()) * 2
+        assert (sum(self.n_callbacks.values()) ==
+                len(f.maker.fgraph.toposort()) * 2)
 
     def test_callback_with_ifelse(self):
         a, b, c = tensor.scalars('abc')
@@ -88,8 +84,8 @@ def test_speed():
         t_b = t3 - t2
 
         print "%s takes %f s/Kop" % (
-                'numpy',
-                (1000*(t_b-t_a) / (steps_b - steps_a)))
+            'numpy',
+            (1000*(t_b-t_a) / (steps_b - steps_a)))
 
     def time_linker(name, linker):
         steps_a = 5
@@ -99,13 +95,9 @@ def test_speed():
         b = build_graph(x, steps_b)
 
         f_a = function([x], a,
-                       mode=Mode(optimizer=None, linker=linker()),
-                       #profile='f_a speed test %s'%name,
-        )
+                       mode=Mode(optimizer=None, linker=linker()))
         f_b = function([x], b,
-                       mode=Mode(optimizer=None, linker=linker()),
-                       #profile='f_b speed test %s'%name,
-        )
+                       mode=Mode(optimizer=None, linker=linker()))
 
         f_a([2.0, 3.0])
         t0 = time.time()
@@ -151,14 +143,10 @@ def test_speed_lazy():
 
         f_a = function([x], a,
                        mode=Mode(optimizer=None,
-                                 linker=linker()),
-                       #profile='f_a lazy ifelse %s'%name,
-        )
+                                 linker=linker()))
         f_b = function([x], b,
                        mode=Mode(optimizer=None,
-                                 linker=linker()),
-                       #profile='f_b lazy ifelse %s'%name,
-        )
+                                 linker=linker()))
 
         f_a([2.0])
         t0 = time.time()
@@ -175,8 +163,8 @@ def test_speed_lazy():
         t_b = t3 - t2
 
         print "%s takes %f s/Kop" % (
-                name,
-                (1000*(t_b-t_a) / (steps_b - steps_a)))
+            name,
+            (1000*(t_b-t_a) / (steps_b - steps_a)))
 
     time_linker('vmLinker', vm.VM_Linker)
     time_linker('vmLinker_nogc', lambda: vm.VM_Linker(allow_gc=False))
@@ -218,7 +206,7 @@ if run_memory_usage_tests:
         for i in xrange(1000000):
             n = numpy.asarray([2.3, 4.5], dtype='f')
             c = sys.getrefcount(n)
-            a = cuda.CudaNdarray(n)
+            cuda.CudaNdarray(n)
             assert c == sys.getrefcount(n)
             if not i % 1000:
                 print '.',
@@ -258,7 +246,7 @@ if run_memory_usage_tests:
         def build_graph(x, depth=5):
             z = x
             for d in range(depth):
-                z = ifelse(z.mean()>0.5, -z, z)
+                z = ifelse(z.mean() > 0.5, -z, z)
             return z
 
         def time_linker(name, linker):

theano/tests/test_flake8.py

@@ -320,7 +320,6 @@ whitelist_flake8 = [
     "gof/tests/test_destroyhandler.py",
     "gof/tests/test_opt.py",
     "gof/tests/test_lazy.py",
-    "gof/tests/test_vm.py",
     "gof/tests/test_toolbox.py",
     "gof/tests/test_link.py",
     "gof/tests/test_fg.py",