diff --git a/tests/loss/test_barlow_twins_loss.py b/tests/loss/test_barlow_twins_loss.py index 5ef4de3da..b25246132 100644 --- a/tests/loss/test_barlow_twins_loss.py +++ b/tests/loss/test_barlow_twins_loss.py @@ -91,4 +91,4 @@ def test__loss_is_affine_invariant(self) -> None: x = torch.randn(32, 1024) # Loss should be invariant to affine transformations. - assert torch.allclose(loss(x, x), loss(x, 2 * x + 4)) + assert torch.allclose(loss(x, x), loss(x, 2*x + 4))