diff --git a/tests/plots/interactive_plotter.py b/tests/plots/interactive_plotter.py index a722ccc1..1c2b6240 100644 --- a/tests/plots/interactive_plotter.py +++ b/tests/plots/interactive_plotter.py @@ -61,7 +61,7 @@ def main() -> None: "AlignedMTL-RMSE": lambda: AlignedMTL(scale_mode="rmse"), str(CAGrad(c=0.5)): lambda: CAGrad(c=0.5), str(ConFIG()): lambda: ConFIG(), - str(DualProj()): lambda: DualProj(), + str(DualProj()): lambda: DualProj(reg_eps=1e-7), str(GradDrop()): lambda: GradDrop(), str(GradVac()): lambda: GradVac(), str(IMTLG()): lambda: IMTLG(), @@ -72,7 +72,7 @@ def main() -> None: str(Random()): lambda: Random(), str(Sum()): lambda: Sum(), str(TrimmedMean(trim_number=1)): lambda: TrimmedMean(trim_number=1), - str(UPGrad()): lambda: UPGrad(), + str(UPGrad()): lambda: UPGrad(reg_eps=1e-7), } aggregator_strings = list(aggregator_factories.keys())