[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

src/torchjd/aggregation/_stch.py

@@ -190,9 +190,9 @@ def forward(self, gramian: PSDMatrix) -> Tensor:
                 if self.nadir_accumulator is None:
                     self.nadir_accumulator = grad_norms.detach().clone()
                 else:
-                    self.nadir_accumulator = self.nadir_accumulator.to(
-                        device=device, dtype=dtype
-                    ) + grad_norms.detach()
+                    self.nadir_accumulator = (
+                        self.nadir_accumulator.to(device=device, dtype=dtype) + grad_norms.detach()
+                    )
 
                 self.step += 1