From 5e9a9838a21b34a3c3ff0781328442140b8b5ab8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Val=C3=A9rian=20Rey?= Date: Sun, 23 Mar 2025 11:31:04 +0100 Subject: [PATCH] `GradDrop` error message typo --- src/torchjd/aggregation/graddrop.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/torchjd/aggregation/graddrop.py b/src/torchjd/aggregation/graddrop.py index acf0cd70c..97d671e76 100644 --- a/src/torchjd/aggregation/graddrop.py +++ b/src/torchjd/aggregation/graddrop.py @@ -72,8 +72,8 @@ def _check_matrix_has_enough_rows(self, matrix: Tensor) -> None: n_rows = matrix.shape[0] if self.leak is not None and n_rows != len(self.leak): raise ValueError( - f"Parameter `matrix` should be a matrix of at least {len(self.leak)} rows " - f"(i.e. the number of leak scalars). Found `matrix` of shape `{matrix.shape}`." + f"Parameter `matrix` should be a matrix of exactly {len(self.leak)} rows (i.e. the " + f"number of leak scalars). Found `matrix` of shape `{matrix.shape}`." ) def __repr__(self) -> str: