Blob Blame History Raw
diff -up theano/tensor/tests/test_sort.py.orig theano/tensor/tests/test_sort.py
--- theano/tensor/tests/test_sort.py.orig	2019-08-23 20:46:06.243192142 +0200
+++ theano/tensor/tests/test_sort.py	2019-08-23 20:45:55.490214412 +0200
@@ -37,7 +37,7 @@ class Test_sort(unittest.TestCase):
 
     def test2(self):
         a = tensor.dmatrix()
-        axis = tensor.scalar()
+        axis = tensor.scalar(dtype="int64")
         w = sort(a, axis)
         f = theano.function([a, axis], w)
         for axis_val in 0, 1:
@@ -55,7 +55,7 @@ class Test_sort(unittest.TestCase):
 
     def test4(self):
         a = tensor.dmatrix()
-        axis = tensor.scalar()
+        axis = tensor.scalar(dtype="int64")
         l = sort(a, axis, "mergesort")
         f = theano.function([a, axis], l)
         for axis_val in 0, 1: