Fix codacy

Author: FilippoOlivo

pina/callback/refinement/__init__.py

@@ -1,3 +1,7 @@
+"""
+Module for Pina Refinement callbacks.
+"""
+
 __all__ = [
     "RefinementInterface",
     "R3Refinement",

pina/callback/refinement/r3_refinement.py

@@ -1,7 +1,7 @@
 """Module for the R3Refinement callback."""
 
 import torch
-import torch.nn as nn
+from torch import nn
 from torch.nn.modules.loss import _Loss
 from .refinement_interface import RefinementInterface
 from ...label_tensor import LabelTensor
@@ -66,5 +66,4 @@ def sample(self, current_points, condition_name, solver):
             retain_pts = len(pts)
             samples = domain.sample(num_old_points - retain_pts, "random")
             return LabelTensor.cat([pts, samples])
-        else:
-            return domain.sample(num_old_points, "random")
+        return domain.sample(num_old_points, "random")

pina/callback/refinement/refinement_interface.py

@@ -7,7 +7,6 @@
 from lightning.pytorch import Callback
 from ...utils import check_consistency
 from ...solver.physics_informed_solver import PINNInterface
-from ...condition import DomainEquationCondition
 
 
 class RefinementInterface(Callback, metaclass=ABCMeta):
@@ -46,7 +45,8 @@ def on_train_start(self, trainer, solver):
         # check we have valid conditions names
         if self._condition_to_update is None:
             self._condition_to_update = [
-                name for name, cond in solver.problem.conditions.items()
+                name
+                for name, cond in solver.problem.conditions.items()
                 if hasattr(cond, "domain")
             ]
 
@@ -64,7 +64,7 @@ def on_train_start(self, trainer, solver):
         # check solver
         if not isinstance(solver, PINNInterface):
             raise RuntimeError(
-                f"Refinment strategies are currently implemented only "
+                "Refinment strategies are currently implemented only "
                 "for physics informed based solvers. Please use a Solver "
                 "inheriting from 'PINNInterface'."
             )
@@ -92,7 +92,6 @@ def sample(self, current_points, condition_name, solver):
         """
         Samples new points based on the condition.
         """
-        pass
 
     @property
     def dataset(self):