From 99b80d5599237b7b199c6007de8b15d33e9a9d3b Mon Sep 17 00:00:00 2001 From: Ryan Soklaski Date: Sun, 8 Sep 2024 13:45:57 -0400 Subject: [PATCH] fix flakes --- tests/tensor_ops/test_sequence_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/tensor_ops/test_sequence_ops.py b/tests/tensor_ops/test_sequence_ops.py index 0d0f4b45..50e3be0d 100644 --- a/tests/tensor_ops/test_sequence_ops.py +++ b/tests/tensor_ops/test_sequence_ops.py @@ -58,7 +58,7 @@ def test_sequential_arithmetic( f = mygrad_func(*tensors) f1 = true_func(tensors_copy) - assert_allclose(f.data, f1.data, rtol=1e-05) + assert_allclose(f.data, f1.data, rtol=1e-05, atol=1e-05) f.sum().backward() f1.sum().backward()