diff --git a/tests/protein/tensor/test_angles.py b/tests/protein/tensor/test_angles.py index a6fb4a53..2f56b385 100644 --- a/tests/protein/tensor/test_angles.py +++ b/tests/protein/tensor/test_angles.py @@ -126,7 +126,7 @@ def test_dihedrals_to_rad(): delta[delta.nonzero()] = torch.abs(delta[torch.nonzero(delta)] - 2 * np.pi) delta = ((delta + 2 * np.pi) / np.pi) % 2 - np.testing.assert_allclose(delta, torch.zeros_like(delta), atol=1e-5) + np.testing.assert_allclose(delta, torch.zeros_like(delta), atol=1e-4, rtol=1e-4) @pytest.mark.skipif(not TORCH_AVAIL, reason="PyTorch not available")