diff --git a/src/torchjd/_linalg/_dual_cone.py b/src/torchjd/_linalg/_dual_cone.py index ff4bc692..69a86689 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 d33cbf3b..8d646754 100644 --- a/src/torchjd/aggregation/_gradvac.py +++ b/src/torchjd/aggregation/_gradvac.py @@ -189,4 +189,4 @@ def reset(self) -> None: self.gramian_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})"