flake8 fixes

theano/sandbox/multinomial.py

@@ -192,6 +192,7 @@ class MultinomialFromUniform(Op):
                             z[0][n, m] += 1
                             break
 
+
 class WeightedSelectionFromUniform(Op):
     """
     Converts samples from a uniform into sample from a multinomial.
@@ -235,11 +236,11 @@ class WeightedSelectionFromUniform(Op):
         if n_samples > pvals.shape[1]:
             raise ValueError("Cannot sample without replacement n samples bigger "
                              "than the size of the distribution.")
-        
+
         if unis.shape[0] != pvals.shape[0] * n_samples:
             raise ValueError("unis.shape[0] != pvals.shape[0] * n_samples",
                              unis.shape[0], pvals.shape[0], n_samples)
-        
+
         if z[0] is None or not numpy.all(z[0].shape == [pvals.shape[0], n_samples]):
             z[0] = -1 * numpy.ones((pvals.shape[0], n_samples), dtype='int')
 
@@ -251,7 +252,7 @@ class WeightedSelectionFromUniform(Op):
         for c in range(n_samples):
             for n in range(nb_multi):
                 cummul = 0
-                unis_n = unis[c*nb_multi+n]
+                unis_n = unis[c * nb_multi + n]
                 for m in range(nb_outcomes):
                     cummul += pvals[n, m]
                     if (cummul > unis_n):
@@ -261,6 +262,7 @@ class WeightedSelectionFromUniform(Op):
                         pvals[n] /= pvals[n].sum()
                         break
 
+
 class GpuMultinomialFromUniform(MultinomialFromUniform, GpuOp):
     """
     The output is transposed compared to MultinomialFromUniform.

theano/sandbox/tests/test_weighted_select.py

 import numpy
-import theano
 from theano import config, function, tensor
 from theano.sandbox import multinomial
 from theano.sandbox.rng_mrg import MRG_RandomStreams as RandomStreams
 import unittest
 
+
 class test_OP(unittest.TestCase):
 
     def test_select_distinct(self):
@@ -22,7 +22,7 @@ class test_OP(unittest.TestCase):
         numpy.random.seed(12345)
         for i in [5, 10, 50, 100, 500, n_elements]:
             uni = numpy.random.rand(i).astype(config.floatX)
-            pvals = numpy.random.randint(1,100,(1,n_elements)).astype(config.floatX)
+            pvals = numpy.random.randint(1, 100, (1, n_elements)).astype(config.floatX)
             pvals /= pvals.sum(1)
             res = f(pvals, uni, i)
             res = numpy.squeeze(res)
@@ -45,7 +45,7 @@ class test_OP(unittest.TestCase):
         n_selected = 200
         numpy.random.seed(12345)
         uni = numpy.random.rand(n_selected).astype(config.floatX)
-        pvals = numpy.random.randint(1,100,(1,n_elements)).astype(config.floatX)
+        pvals = numpy.random.randint(1, 100, (1, n_elements)).astype(config.floatX)
         pvals /= pvals.sum(1)
         self.assertRaises(ValueError, f, pvals, uni, n_selected)
 
@@ -65,7 +65,7 @@ class test_OP(unittest.TestCase):
         n_selected = 10
         mean_rtol = 0.04
         numpy.random.seed(12345)
-        pvals = numpy.random.randint(1,100,(1,n_elements)).astype(config.floatX)
+        pvals = numpy.random.randint(1, 100, (1, n_elements)).astype(config.floatX)
         pvals /= pvals.sum(1)
         avg_pvals = numpy.zeros((n_elements,))
 
@@ -95,7 +95,7 @@ class test_function(unittest.TestCase):
         n_elements = 1000
         numpy.random.seed(12345)
         for i in [5, 10, 50, 100, 500, n_elements]:
-            pvals = numpy.random.randint(1,100,(1,n_elements)).astype(config.floatX)
+            pvals = numpy.random.randint(1, 100, (1, n_elements)).astype(config.floatX)
             pvals /= pvals.sum(1)
             res = f(pvals, i)
             res = numpy.squeeze(res)
@@ -118,7 +118,7 @@ class test_function(unittest.TestCase):
         n_elements = 100
         n_selected = 200
         numpy.random.seed(12345)
-        pvals = numpy.random.randint(1,100,(1,n_elements)).astype(config.floatX)
+        pvals = numpy.random.randint(1, 100, (1, n_elements)).astype(config.floatX)
         pvals /= pvals.sum(1)
         self.assertRaises(ValueError, f, pvals, n_selected)
 
@@ -139,7 +139,7 @@ class test_function(unittest.TestCase):
         n_selected = 10
         mean_rtol = 0.04
         numpy.random.seed(12345)
-        pvals = numpy.random.randint(1,100,(1,n_elements)).astype(config.floatX)
+        pvals = numpy.random.randint(1, 100, (1, n_elements)).astype(config.floatX)
         pvals /= pvals.sum(1)
         avg_pvals = numpy.zeros((n_elements,))