rng_mrg - fixed mistake calling infer_ndim_bcast in multinomial

theano/sandbox/rng_mrg.py

@@ -755,7 +755,8 @@ class MRG_RandomStreams(object):
         pvals = as_tensor_variable(pvals)
         if n == 1 and pvals.ndim == 2:
             ndim, size, bcast = raw_random._infer_ndim_bcast(
-                    ndim, size, n, pvals[:,0])
+                    ndim, size, pvals[:,0])
+            assert ndim==1
             bcast = bcast+(pvals.type.broadcastable[-1],)
             unis = self.uniform(size=size, ndim=1)
             op = multinomial.Multinomial(dtype)

theano/sandbox/test_rng_mrg.py

@@ -555,7 +555,7 @@ def test_multinomial():
 
     sys.stdout.flush()
 
-    if mode!='FAST_COMPILE' and cuda_enabled:
+    if mode != 'FAST_COMPILE' and cuda_enabled:
         print ''
         print 'ON GPU:'
         R = MRG_RandomStreams(234, use_cuda=True)
@@ -571,3 +571,4 @@ def test_multinomial():
         sys.stdout.flush()
         basic_multinomialtest(f, steps, sample_size, pvals, prefix='gpu mrg ')
         numpy.testing.assert_array_almost_equal(out, gpu_out, decimal=6)
+