flake8 for theano/misc/pycuda_example.py; 2 E left

theano/misc/pycuda_example.py

@@ -33,16 +33,14 @@ from theano.sandbox.cuda.basic_ops import (as_cuda_ndarray_variable,
 from theano.sandbox.cuda.opt import gpu_seqopt
 from theano.tensor.utils import hash_from_dict
 
-from . import pycuda_init
-if not pycuda_init.pycuda_available:
-    raise Exception("No pycuda available. You can't load pycuda_example.py")
-
 import pycuda
-from pycuda.elementwise import ElementwiseKernel
 from pycuda.compiler import SourceModule
-from pycuda.tools import VectorArg
 import pycuda.gpuarray
 
+from . import pycuda_init
+if not pycuda_init.pycuda_available:
+    raise Exception("No pycuda available. You can't load pycuda_example.py")
+
 
 def _replace_npy_types(c_arg):
     c_arg = c_arg.replace('npy_float32', 'float')
@@ -235,12 +233,11 @@ class PycudaElemwiseSourceModuleOp(GpuOp):
         c_code = self.scalar_op.c_code(out_node, "some_name",
                                        tuple([n + "[i]" for n in in_name]),
                                        tuple(n + "[i]" for n in out_name), {})
-        c_code_param = ", ".join([
-            _replace_npy_types(var.type.dtype_specs()[1]) + " *" + name
-                               for var, name in chain(izip(inputs, in_name),
-                                                      izip(out_node.outputs,
-                                                           out_name))
-        ] + ["int size"])
+        c_code_param = ", ".join(
+            [_replace_npy_types(var.type.dtype_specs()[1]) + " *" + name
+             for var, name in chain(izip(inputs, in_name),
+                                    izip(out_node.outputs, out_name))] +
+            ["int size"])
         mod = SourceModule("""
   __global__ void %s(%s)
   {
@@ -259,8 +256,8 @@ class PycudaElemwiseSourceModuleOp(GpuOp):
         # TODO assert all input have the same shape
         z, = out
         if (z[0] is None or
-            z[0].shape != inputs[0].shape or
-            not z[0].is_c_contiguous()):
+                z[0].shape != inputs[0].shape or
+                not z[0].is_c_contiguous()):
             z[0] = theano.sandbox.cuda.CudaNdarray.zeros(inputs[0].shape)
         if inputs[0].shape != inputs[1].shape:
             raise TypeError("PycudaElemwiseSourceModuleOp:"
@@ -329,8 +326,8 @@ class PycudaElemwiseSourceModuleMakeThunkOp(Op):
         c_code_param = ", ".join(
             [_replace_npy_types(var.type.dtype_specs()[1]) + " *" + name
              for var, name in chain(izip(node.inputs, in_name),
-                                    izip(node.outputs, out_name))]
-            + ["int size"])
+                                    izip(node.outputs, out_name))] +
+            ["int size"])
 
         mod = SourceModule("""
   __global__ void %s(%s)
@@ -349,8 +346,8 @@ class PycudaElemwiseSourceModuleMakeThunkOp(Op):
         def thunk():
             z = outputs[0]
             if (z[0] is None or
-                z[0].shape != inputs[0][0].shape or
-                not z[0].is_c_contiguous()):
+                    z[0].shape != inputs[0][0].shape or
+                    not z[0].is_c_contiguous()):
                 z[0] = theano.sandbox.cuda.CudaNdarray.zeros(
                     inputs[0][0].shape)
             if inputs[0][0].shape != inputs[1][0].shape:
@@ -384,7 +381,7 @@ def local_pycuda_gpu_elemwise(node):
     """
     if isinstance(node.op, GpuElemwise):
         if (not any([any(i.type.broadcastable) for i in node.inputs]) and
-            all([i.ndim <= 2 for i in node.inputs])):
+                all([i.ndim <= 2 for i in node.inputs])):
             new_op = PycudaElemwiseSourceModuleOp(node.op.scalar_op,
                                                   node.op.inplace_pattern)(
                                                       *node.inputs)

theano/tests/test_flake8.py

@@ -184,7 +184,6 @@ whitelist_flake8 = [
     "scan_module/scan_opt.py",
     "scan_module/tests/test_scan.py",
     "scan_module/tests/test_scan_opt.py",
-    "misc/pycuda_example.py",
     "misc/ordered_set.py",
     "misc/strutil.py",
     "misc/gnumpy_utils.py",