fix(aggregation): GradDrop error message typo (#262)

ValerianRey · web-flow · commit 40842c4154b6 · 2025-03-23T11:23:07.000+01:00
diff --git a/src/torchjd/aggregation/graddrop.py b/src/torchjd/aggregation/graddrop.py
@@ -39,7 +39,7 @@ class GradDrop(Aggregator):
     def __init__(self, f: Callable = _identity, leak: Tensor | None = None):
         if leak is not None and leak.dim() != 1:
             raise ValueError(
-                "Parameter `leak` should be a 1-dimensional tensor. Found `weights.shape = "
+                "Parameter `leak` should be a 1-dimensional tensor. Found `leak.shape = "
                 f"{leak.shape}`."
             )
 

Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,7 @@ class GradDrop(Aggregator):`
`39`	`39`	`def __init__(self, f: Callable = _identity, leak: Tensor \| None = None):`
`40`	`40`	`if leak is not None and leak.dim() != 1:`
`41`	`41`	`raise ValueError(`
`42`		- "Parameter `leak` should be a 1-dimensional tensor. Found `weights.shape = "
	`42`	+ "Parameter `leak` should be a 1-dimensional tensor. Found `leak.shape = "
`43`	`43`	f"{leak.shape}`."
`44`	`44`	`)`
`45`	`45`