From 5dea4f843c80be477a556ebe76de13332beba5e8 Mon Sep 17 00:00:00 2001 From: xinyu-intel Date: Mon, 2 Sep 2019 22:24:17 +0800 Subject: [PATCH] fix flaky test --- tests/python/quantization/test_quantization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/quantization/test_quantization.py b/tests/python/quantization/test_quantization.py index d753b514302e..0a9fcda56a64 100644 --- a/tests/python/quantization/test_quantization.py +++ b/tests/python/quantization/test_quantization.py @@ -265,7 +265,7 @@ def check_quantized_conv(data_shape, kernel, num_filter, pad, stride, no_bias, q qoutput, min_range, max_range = conv_exe_int8.forward() if no_bias: - assert_almost_equal(output.asnumpy(), qoutput.asnumpy()) + assert_almost_equal(output.asnumpy(), qoutput.asnumpy(), atol = 1) else: # with adding bias, accuracy loss should not be greater than one diff = mx.nd.abs(output - qoutput.astype(output.dtype))