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