black formatter

This commit is contained in:
giovanni
2025-03-14 00:19:18 +01:00
committed by Nicola Demo
parent d2e3f458ab
commit c164db874b
8 changed files with 10 additions and 9 deletions

View File

@@ -40,7 +40,7 @@ class GAROM(MultiSolverInterface):
):
"""
Initialization of the :class:`GAROM` class.
:param AbstractProblem problem: The formulation of the problem.
:param torch.nn.Module generator: The generator model.
:param torch.nn.Module discriminator: The discriminator model.
@@ -50,7 +50,7 @@ class GAROM(MultiSolverInterface):
:param Optimizer optimizer_generator: The optimizer for the generator.
If `None`, the :class:`torch.optim.Adam` optimizer is used.
Default is ``None``.
:param Optimizer optimizer_discriminator: The optimizer for the
:param Optimizer optimizer_discriminator: The optimizer for the
discriminator. If `None`, the :class:`torch.optim.Adam` optimizer is
used. Default is ``None``.
:param Scheduler scheduler_generator: The learning rate scheduler for