From 5a832c2f0254d7825d72278b4bbb5b19b5021421 Mon Sep 17 00:00:00 2001 From: PhiSpel <62760840+PhiSpel@users.noreply.github.com> Date: Wed, 11 Sep 2024 10:58:53 +0200 Subject: [PATCH] Update test_einsum.py --- tests/flow/test_einsum.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/flow/test_einsum.py b/tests/flow/test_einsum.py index 09c059b6..c93c77fc 100644 --- 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()