From e51b2fc8b84e8d1632f49402c48e78d4eea509be Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Val=C3=A9rian=20Rey?= Date: Wed, 7 Oct 2026 17:58:13 +0200 Subject: [PATCH] Use class name in GradVac and QuadprogProjector repr GradVac.__repr__ and QuadprogProjector.__repr__ hard-coded the class name. They now use `self.__class__.__name__`, like the other classes, so that subclasses get their own name. Co-Authored-By: Claude Opus 5.5 --- src/torchjd/_linalg/_dual_cone.py | 2 +- src/torchjd/aggregation/_gradvac.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/torchjd/_linalg/_dual_cone.py b/src/torchjd/_linalg/_dual_cone.py index 7dd32482..01e80f56 100644 --- a/src/torchjd/_linalg/_dual_cone.py +++ b/src/torchjd/_linalg/_dual_cone.py @@ -102,7 +102,7 @@ def reg_eps(self, value: float) -> None: self._reg_eps = value def __repr__(self) -> str: - return f"QuadprogProjector(norm_eps={self._norm_eps}, reg_eps={self._reg_eps})" + return f"{self.__class__.__name__}(norm_eps={self._norm_eps}, reg_eps={self._reg_eps})" def __call__(self, U: Tensor, G: PSDMatrix) -> Tensor: diff --git a/src/torchjd/aggregation/_gradvac.py b/src/torchjd/aggregation/_gradvac.py index 73464c11..4ef537e0 100644 --- a/src/torchjd/aggregation/_gradvac.py +++ b/src/torchjd/aggregation/_gradvac.py @@ -191,4 +191,4 @@ def reset(self) -> None: self._gradvac_weighting.reset() def __repr__(self) -> str: - return f"GradVac(beta={self.beta!r}, eps={self.eps!r})" + return f"{self.__class__.__name__}(beta={self.beta!r}, eps={self.eps!r})"