diff --git a/tests/dynamical/dynamical_fixtures.py b/tests/dynamical/dynamical_fixtures.py index c0123af9..27046405 100644 --- a/tests/dynamical/dynamical_fixtures.py +++ b/tests/dynamical/dynamical_fixtures.py @@ -103,7 +103,7 @@ def check_states_match(state1: State[torch.Tensor], state2: State[torch.Tensor]) for k in state1.keys(): assert torch.allclose( - state1[k], state2[k], atol=1e-6, rtol=1e-3 + state1[k], state2[k], # atol=1e-6, rtol=1e-3 ), f"Trajectories differ in state trajectory of variable {k}, but should be identical." return True