black formatter

pina/solver/garom.py

@@ -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

pina/solver/solver.py

@@ -63,7 +63,7 @@ class SolverInterface(lightning.pytorch.LightningModule, metaclass=ABCMeta):
     def _check_solver_consistency(self, problem):
         """
         Check the consistency of the solver with the problem formulation.
-        
+
         :param AbstractProblem problem: The problem to be solved.
         """
         for condition in problem.conditions.values():
@@ -290,6 +290,7 @@ class SingleSolverInterface(SolverInterface, metaclass=ABCMeta):
     """
     Base class for PINA solvers using a single :class:`torch.nn.Module`.
     """
+
     def __init__(
         self,
         problem,