Blame python-theano-sort.patch
|
 |
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:
|