[incubator-mxnet] branch master updated: fix test_stn (#14063)

[ke...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

kellen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new d02beeb  fix test_stn (#14063)
d02beeb is described below

commit d02beeb6e4bad35a046810baac759ae50e7fdceb
Author: Sheng Zha <sz...@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 146836c..6489756 100644
--- a/tests/python/unittest/test_operator.py
+++ b/tests/python/unittest/test_operator.py
@@ -2530,7 +2530,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