diff --git a/src/torchjd/aggregation/_dualproj.py b/src/torchjd/aggregation/_dualproj.py index 84206b47..a9a4172a 100644 --- a/src/torchjd/aggregation/_dualproj.py +++ b/src/torchjd/aggregation/_dualproj.py @@ -29,7 +29,7 @@ def __init__( ) -> None: super().__init__() self.pref_vector = pref_vector - self.projector = projector_or_default(projector) + self.projector = projector def forward(self, gramian: PSDMatrix, /) -> Tensor: u = self.weighting(gramian) diff --git a/src/torchjd/aggregation/_upgrad.py b/src/torchjd/aggregation/_upgrad.py index 3ac87695..2e12e84d 100644 --- a/src/torchjd/aggregation/_upgrad.py +++ b/src/torchjd/aggregation/_upgrad.py @@ -30,7 +30,7 @@ def __init__( ) -> None: super().__init__() self.pref_vector = pref_vector - self.projector = projector_or_default(projector) + self.projector = projector def forward(self, gramian: PSDMatrix, /) -> Tensor: U = torch.diag(self.weighting(gramian))