[TEST] Address flaky error in test_any (apache#6705)

trevor-m · Dec 2, 2020 · 9478123 · 9478123
1 parent f2f9d20
commit 9478123
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):