Flake8 for nnet.py

theano/sandbox/gpuarray/nnet.py

@@ -5,9 +5,11 @@ import os
 from theano import Op, Apply, config
 from six import StringIO
 
+from theano.gof.util import MethodNotDefined
+
 try:
     import pygpu
-    from pygpu import gpuarray, elemwise
+    from pygpu import gpuarray
 except ImportError:
     pass
 
@@ -665,15 +667,15 @@ class GpuSoftmax(GpuKernelBase, Op):
             ]
         kernels = []
         kname = "kSoftmax"
-        k_var= "kSoftmax_" + nodename
-        code = nvcc_kernel(kname,
+        k_var = "kSoftmax_" + nodename
+        code = nvcc_kernel(
+            kname,
             params=['const ga_size M', 'const ga_size N',
                     'const %s * x' % type_x, 'const ga_size offset_x',
                     'const ga_ssize sx0', 'const ga_ssize sx1',
                     '%s * sm' % type_sm, 'const ga_size offset_sm',
                     'const ga_ssize sm_s0', 'const ga_ssize sm_s1'],
-                body=[
-                    "extern __shared__ %s buf[]" % type_acc,
+            body=["extern __shared__ %s buf[]" % type_acc,
                   "%s * buf2 = buf + N" % type_acc,
                   "x = (const %s *)(((char *)x)+offset_x)" % type_x,
                   "sm = (%s *)(((char *)sm)+offset_sm)" % type_sm,
@@ -696,15 +698,15 @@ class GpuSoftmax(GpuKernelBase, Op):
         kernels.append(Kernel(code=code, name=kname, params=params,
                               flags=flags, objvar=k_var))
         kname = "kSoftmax_fixed_shared"
-        k_var= "kSoftmax_fixed_shared" + nodename
-        code = nvcc_kernel(kname,
+        k_var = "kSoftmax_fixed_shared" + nodename
+        code = nvcc_kernel(
+            kname,
             params=['const ga_size M', 'const ga_size N',
                     'const %s * x' % type_x, 'const ga_size offset_x',
                     'const ga_ssize sx0', 'const ga_ssize sx1',
                     '%s * sm' % type_sm, 'const ga_size offset_sm',
                     'const ga_ssize sm_s0', 'const ga_ssize sm_s1'],
-                body=[
-                    "extern __shared__ %s buf[]" % type_acc,
+            body=["extern __shared__ %s buf[]" % type_acc,
                   "x = (const %s *)(((char *)x)+offset_x)" % type_x,
                   "sm = (%s *)(((char *)sm)+offset_sm)" % type_sm,
                   "for (int blockIDX = blockIdx.x; blockIDX < M;"
@@ -892,7 +894,8 @@ class GpuSoftmaxWithBias (GpuKernelBase, Op):
         kernels = []
         kname = "kSoftmaxWithBias"
         k_var = "kSoftmaxWithBias_" + nodename
-        code = nvcc_kernel(kname,
+        code = nvcc_kernel(
+            kname,
             params=['const ga_size M', 'const ga_size N',
                     'const %s * x' % type_x, 'const ga_size offset_x',
                     'const ga_ssize sx0', 'const ga_ssize sx1',
@@ -900,8 +903,7 @@ class GpuSoftmaxWithBias (GpuKernelBase, Op):
                     'const ga_ssize sb0',
                     '%s * sm' % type_sm, 'const ga_size offset_sm',
                     'const ga_ssize sm_s0', 'const ga_ssize sm_s1'],
-                body=[
-                    "extern __shared__ %s buf[]" % type_acc,
+            body=["extern __shared__ %s buf[]" % type_acc,
                   "%s * buf2 = buf + N" % type_acc,
                   "x = (const %s *)(((char *)x)+offset_x)" % type_x,
                   "b = (const %s *)(((char *)b)+offset_b)" % type_b,
@@ -926,7 +928,8 @@ class GpuSoftmaxWithBias (GpuKernelBase, Op):
                               flags=flags, objvar=k_var))
         kname = "kSoftmaxWithBias_fixed_shared"
         k_var = "kSoftmaxWithBias_fixed_shared" + nodename
-        code = nvcc_kernel(kname,
+        code = nvcc_kernel(
+            kname,
             params=['const ga_size M', 'const ga_size N',
                     'const %s * x' % type_x, 'const ga_size offset_x',
                     'const ga_ssize sx0', 'const ga_ssize sx1',
@@ -934,8 +937,7 @@ class GpuSoftmaxWithBias (GpuKernelBase, Op):
                     'const ga_ssize sb0',
                     '%s * sm' % type_sm, 'const ga_size offset_sm',
                     'const ga_ssize sm_s0', 'const ga_ssize sm_s1'],
-                body=[
-                    "extern __shared__ %s buf[]" % type_acc,
+            body=["extern __shared__ %s buf[]" % type_acc,
                   "x = (const %s *)(((char *)x)+offset_x)" % type_x,
                   "b = (const %s *)(((char *)b)+offset_b)" % type_b,
                   "sm = (%s *)(((char *)sm)+offset_sm)" % type_sm,

theano/tests/test_flake8.py

@@ -157,7 +157,6 @@ whitelist_flake8 = [
     "sandbox/linalg/ops.py",
     "sandbox/linalg/__init__.py",
     "sandbox/linalg/tests/test_linalg.py",
-    "sandbox/gpuarray/nnet.py",
     "sandbox/gpuarray/elemwise.py",
     "sandbox/gpuarray/type.py",
     "sandbox/gpuarray/__init__.py",