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: