diff --git a/test/test_ops.py b/test/test_ops.py index 06195fa8..e110c761 100644 --- a/test/test_ops.py +++ b/test/test_ops.py @@ -39,7 +39,7 @@ class TestOps(unittest.TestCase): helper_test_op([(45,65), (45,65)], lambda x,y: x*y, Tensor.mul) def test_div(self): # TODO: why does this need more tolerance? - helper_test_op([(45,65), (45,65)], lambda x,y: x/y, Tensor.div, atol=5e-5) + helper_test_op([(45,65), (45,65)], lambda x,y: x/y, Tensor.div, atol=5e-5, grad_atol=1e-5) def test_pow(self): helper_test_op([(45,65), (45,65)], lambda x,y: x**y, Tensor.pow) def test_sqrt(self):