Skip to content

Commit 66ca700

Browse files
authored
fix(aggregation): GradDrop error message typo (#263)
1 parent 40842c4 commit 66ca700

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/torchjd/aggregation/graddrop.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,8 +72,8 @@ def _check_matrix_has_enough_rows(self, matrix: Tensor) -> None:
7272
n_rows = matrix.shape[0]
7373
if self.leak is not None and n_rows != len(self.leak):
7474
raise ValueError(
75-
f"Parameter `matrix` should be a matrix of at least {len(self.leak)} rows "
76-
f"(i.e. the number of leak scalars). Found `matrix` of shape `{matrix.shape}`."
75+
f"Parameter `matrix` should be a matrix of exactly {len(self.leak)} rows (i.e. the "
76+
f"number of leak scalars). Found `matrix` of shape `{matrix.shape}`."
7777
)
7878

7979
def __repr__(self) -> str:

0 commit comments

Comments
 (0)