Fix formatting and failing UT

Author: adendek

pina/_src/adaptive_function/adaptive_function.py

@@ -2,7 +2,9 @@
 
 import torch
 from pina._src.core.utils import check_consistency
-from pina._src.adaptive_function.adaptive_function_interface import AdaptiveActivationFunctionInterface
+from pina._src.adaptive_function.adaptive_function_interface import (
+    AdaptiveActivationFunctionInterface,
+)
 
 
 class AdaptiveReLU(AdaptiveActivationFunctionInterface):

pina/_src/callback/refinement/r3_refinement.py

@@ -1,7 +1,9 @@
 """Module for the R3Refinement callback."""
 
 import torch
-from pina._src.callback.refinement.refinement_interface import RefinementInterface
+from pina._src.callback.refinement.refinement_interface import (
+    RefinementInterface,
+)
 from pina._src.core.label_tensor import LabelTensor
 from pina._src.core.utils import check_consistency
 from pina._src.loss.loss_interface import LossInterface

pina/_src/callback/refinement/refinement_interface.py

@@ -6,7 +6,9 @@
 from abc import ABCMeta, abstractmethod
 from lightning.pytorch import Callback
 from pina._src.core.utils import check_consistency
-from pina._src.solver.physics_informed_solver.pinn_interface import PINNInterface
+from pina._src.solver.physics_informed_solver.pinn_interface import (
+    PINNInterface,
+)
 
 
 class RefinementInterface(Callback, metaclass=ABCMeta):

pina/_src/condition/condition.py

@@ -1,7 +1,9 @@
 """Module for the Condition class."""
 
 from pina._src.condition.data_condition import DataCondition
-from pina._src.condition.domain_equation_condition import DomainEquationCondition
+from pina._src.condition.domain_equation_condition import (
+    DomainEquationCondition,
+)
 from pina._src.condition.input_equation_condition import InputEquationCondition
 from pina._src.condition.input_target_condition import InputTargetCondition
 

pina/_src/core/operator.py

@@ -1,4 +1,4 @@
-""" Module for vectorized differential operators implementation.
+"""Module for vectorized differential operators implementation.
 
 Differential operators are used to define differential problems and are
 implemented to run efficiently on various accelerators, including CPU, GPU, TPU,

pina/_src/core/trainer.py

@@ -6,8 +6,13 @@
 import lightning
 from pina._src.core.utils import check_consistency, custom_warning_format
 from pina._src.data.data_module import PinaDataModule
-from pina._src.solver.supervised_solver.supervised_solver_interface import SolverInterface
-from pina._src.solver.physics_informed_solver.pinn_interface import PINNInterface
+from pina._src.solver.supervised_solver.supervised_solver_interface import (
+    SolverInterface,
+)
+from pina._src.solver.physics_informed_solver.pinn_interface import (
+    PINNInterface,
+)
+
 # set the warning for compile options
 warnings.formatwarning = custom_warning_format
 warnings.filterwarnings("always", category=UserWarning)

pina/_src/domain/ellipsoid_domain.py

@@ -7,7 +7,6 @@
 from pina._src.core.utils import check_consistency
 
 
-
 class EllipsoidDomain(BaseDomain):
     """
     Implementation of the ellipsoid domain.

pina/_src/domain/intersection.py

@@ -6,7 +6,6 @@
 from pina._src.core.utils import check_consistency
 
 
-
 class Intersection(BaseOperation):
     r"""
     Implementation of the intersection operation defined on a list of domains.

pina/_src/domain/simplex_domain.py

@@ -7,7 +7,6 @@
 from pina._src.core.utils import check_consistency
 
 
-
 class SimplexDomain(BaseDomain):
     """
     Implementation of the simplex domain.

pina/_src/model/block/message_passing/equivariant_graph_neural_operator_block.py

@@ -2,7 +2,9 @@
 
 import torch
 from pina._src.core.utils import check_positive_integer
-from pina._src.model.block.message_passing.en_equivariant_network_block import EnEquivariantNetworkBlock
+from pina._src.model.block.message_passing.en_equivariant_network_block import (
+    EnEquivariantNetworkBlock,
+)
 
 
 class EquivariantGraphNeuralOperatorBlock(torch.nn.Module):