From 9478123ce03cdaede6105659806ce844eae1e5e9 Mon Sep 17 00:00:00 2001 From: Tianqi Chen Date: Sun, 18 Oct 2020 06:09:17 -0700 Subject: [PATCH] [TEST] Address flaky error in test_any (#6705) --- tests/python/relay/test_any.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/relay/test_any.py b/tests/python/relay/test_any.py index c445cd1944009..872728514c3e8 100644 --- a/tests/python/relay/test_any.py +++ b/tests/python/relay/test_any.py @@ -58,7 +58,7 @@ def check_result( if flatten: result = result.flatten() expected = expected.flatten() - tvm.testing.assert_allclose(result, expected) + tvm.testing.assert_allclose(result, expected, atol=2e-6) def verify_any_broadcast(x_shape, y_shape, x_np_shape, y_np_shape, op, np_op):