From 37819c8fad90a90f0828bc65f5804002935f806d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Val=C3=A9rian=20Rey?= <valerian.rey@gmail.com>
Date: Sun, 23 Mar 2025 11:18:07 +0100
Subject: [PATCH] Fix error message typo in GradDrop

---
 src/torchjd/aggregation/graddrop.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/torchjd/aggregation/graddrop.py b/src/torchjd/aggregation/graddrop.py
index c45430247..acf0cd70c 100644
--- 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}`."
             )