diff --git a/test/test.py b/test/test.py index 7dff035cd2..31255ca174 100644 --- a/test/test.py +++ b/test/test.py @@ -33,8 +33,8 @@ def test_pytorch(): return outx.detach().numpy(), x.grad, W.grad for x,y in zip(test_tinygrad(), test_pytorch()): - print(x,y) - np.testing.assert_allclose(x, y, atol=1e-6) - + print(x,y) + np.testing.assert_allclose(x, y, atol=1e-6) +