Fix compilation crash and flake8

theano/gpuarray/sort.py

@@ -113,7 +113,7 @@ class GpuTopKOp(GpuKernelBase, TopKOp):
         param_types = [ga.SIZE] * (ndim - 1)  # dims
         for _ in range(self.return_values + self.return_indices):
             param_types.append(ga.GpuArray)  # dst*
-            param_types.extend([ga.SIZE] * ndim)  # offset
+            param_types.append(ga.SIZE)  # offset
             param_types.extend([ga.SSIZE] * ndim)  # dst*_strides
         param_types.append(ga.SIZE)  # k
         param_types.append(ga.GpuArray)  # src

theano/gpuarray/tests/test_sort.py

@@ -2,7 +2,7 @@ from __future__ import absolute_import, print_function, division
 
 import theano
 import theano.tensor.tests.test_sort
-from .config import mode_with_gpu, test_ctx_name
+from .config import mode_with_gpu
 from ..sort import GpuTopKOp
 from nose.plugins.skip import SkipTest