diff --git a/tests/python/quantization/test_quantization.py b/tests/python/quantization/test_quantization.py index 6d6ba41ab7fb..723873ac6fb3 100644 --- a/tests/python/quantization/test_quantization.py +++ b/tests/python/quantization/test_quantization.py @@ -178,7 +178,7 @@ def check_requantize_with_symbol(shape, min_calib_range=None, max_calib_range=No max_range.asscalar(), min_calib_range=min_calib_range, max_calib_range=max_calib_range) - assert_almost_equal(qdata_int8.asnumpy(), qdata_int8_np) + assert_almost_equal(qdata_int8.asnumpy(), qdata_int8_np, atol = 1) assert_almost_equal(min_output.asnumpy(), np.array([min_output_np])) assert_almost_equal(max_output.asnumpy(), np.array([max_output_np]))