diff --git a/tests/python/unittest/test_numpy_op.py b/tests/python/unittest/test_numpy_op.py index e69ac332b55d..c137b30f89c3 100644 --- a/tests/python/unittest/test_numpy_op.py +++ b/tests/python/unittest/test_numpy_op.py @@ -415,7 +415,6 @@ def is_int(dtype): for axis in ([i for i in range(in_data_dim)] + [(), None]): for itype in ['float16', 'float32', 'float64']: for dtype in ['float16', 'float32', 'float64']: - print(itype, dtype) if is_int(dtype) and not is_int(itype): continue # test gluon @@ -494,7 +493,7 @@ def legalize_shape(shape): for dtype in ['float16', 'float32', 'float64']: if is_int(dtype) and not is_int(itype) or is_int(itype) and is_int(dtype): continue - atol = 1e-4 if itype == 'float16' or dtype == 'float16' else 1e-5 + atol = 3e-4 if itype == 'float16' or dtype == 'float16' else 1e-5 rtol = 1e-2 if itype == 'float16' or dtype == 'float16' else 1e-3 # test gluon test_moment = TestMoment(name, axis=axis, dtype=dtype, keepdims=keepdims, ddof=ddof)