Merge pull request #5406 from chinnadhurai/ccw_numpy

theano/sparse/sandbox/sp2.py

@@ -5,6 +5,7 @@ import theano
 import scipy.sparse
 
 from theano import gof, tensor
+from theano.tensor import discrete_dtypes, float_dtypes
 from theano.tensor.opt import register_specialize
 from theano.sparse.basic import (
     as_sparse_variable, SparseType, add_s_s, neg,
@@ -116,6 +117,11 @@ class Binomial(gof.op.Op):
         n = tensor.as_tensor_variable(n)
         p = tensor.as_tensor_variable(p)
         shape = tensor.as_tensor_variable(shape)
+
+        assert n.dtype in discrete_dtypes
+        assert p.dtype in float_dtypes
+        assert shape.dtype in discrete_dtypes
+
         return gof.Apply(self, [n, p, shape],
                          [SparseType(dtype=self.dtype,
                                      format=self.format)()])

theano/sparse/tests/test_sp2.py

@@ -62,7 +62,7 @@ class PoissonTester(utt.InferShapeTester):
 
 
 class BinomialTester(utt.InferShapeTester):
-    n = tensor.scalar()
+    n = tensor.scalar(dtype='int64')
     p = tensor.scalar()
     shape = tensor.lvector()
     _n = 5