This is an automated email from the ASF dual-hosted git repository.
zhasheng pushed a commit to branch v1.4.x
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/v1.4.x by this push:
new 9ba5119 fix test_stn (#14063)
9ba5119 is described below
commit 9ba51194abb569d69d01a5ccb4be7576e886c2b9
Author: Sheng Zha <s...@users.noreply.github.com>
AuthorDate: Sun Feb 3 19:39:49 2019 -0800
fix test_stn (#14063)
---
tests/python/unittest/test_operator.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/tests/python/unittest/test_operator.py
b/tests/python/unittest/test_operator.py
index 0915739..a20f267 100644
--- a/tests/python/unittest/test_operator.py
+++ b/tests/python/unittest/test_operator.py
@@ -2529,7 +2529,8 @@ def test_flip():
@with_seed()
def test_stn():
- np.set_printoptions(threshold=np.nan)
+ import sys
+ np.set_printoptions(threshold=sys.maxsize)
num_filter = 2 # conv of loc net
kernel = (3, 3) # conv of loc net
num_hidden = 6 # fc of loc net
