Update test_einsum.py

lettucecfd · Sep 11, 2024 · 5a832c2 · 5a832c2
1 parent ac3d10c
commit 5a832c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/flow/test_einsum.py b/tests/flow/test_einsum.py
@@ -200,7 +200,7 @@ def test_einsum(fix_dim):
     f_log = -torch.log(torch.einsum("q...,q...->q...", [flow.f,
                                                         1 / flow.torch_stencil.w]))
     f2 = torch.einsum("q...,q...->...", [flow.f, f_log])
-    assert torch.allclose(f0, f1, rtol=)
+    assert torch.allclose(f0, f1)
     assert torch.allclose(f1, f2)
 
     f0 = flow.pseudo_entropy_global()