Code formating and simplification

theano/sandbox/gpuarray/subtensor.py

@@ -425,14 +425,12 @@ class GpuAdvancedIncSubtensor1(HideC, tensor.AdvancedIncSubtensor1):
             elif y.ndim == x.ndim:
                 assert len(y) == len(idx)
 
-                firstIdxY, firstIdxX = enumerate(idx).next()
-                k = self.getInplElemwiseAdditionKernel(x[firstIdxX],
-                                                       y[firstIdxY])                 
+                k = self.getInplElemwiseAdditionKernel(x[0], y[0])
 
                 for (j, i) in enumerate(idx):
                     k(x[i], y[j], broadcast=False)
             else:
-                nb_dims_to_add = (x[idx[0]].ndim - y.ndim)
+                nb_dims_to_add = (x.ndim - 1) - y.ndim
                 reshaped_y = y.reshape((1,)*nb_dims_to_add + y.shape)
                 k = self.getInplElemwiseAdditionKernel(x[0],
                                                        reshaped_y)
@@ -592,4 +590,4 @@ class GpuAdvancedIncSubtensor1_dev20(GpuAdvancedIncSubtensor1):
                 return;
         }
 
-        """ %locals()
\ No newline at end of file
+        """ %locals()

theano/sandbox/gpuarray/tests/test_subtensor.py

@@ -25,7 +25,7 @@ class G_subtensor(T_subtensor):
                              shared=gpuarray_shared_constructor,
                              sub=GpuSubtensor,
                              inc_sub=GpuIncSubtensor,
-                             adv_incsub1 = GpuAdvancedIncSubtensor1,
+                             adv_incsub1=GpuAdvancedIncSubtensor1,
                              mode=mode_with_gpu,
                              # avoid errors with limited devices
                              dtype='float32',