fix test_stn (apache#14063)

lanking520 · Feb 18, 2019 · 99c0ca9 · 99c0ca9
1 parent 0bbb9b9
commit 99c0ca9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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