🎨 Format Python code with psf/black

2024-10-10 16:46:27 +00:00
parent 53605d5a09
commit 1f6ea277ad
1 changed files with 5 additions and 1 deletions
--- a/pina/solvers/pinns/rbapinn.py
+++ b/pina/solvers/pinns/rbapinn.py
@@ -158,7 +158,11 @@ class RBAPINN(PINN):
        residual = self.compute_residual(samples=samples, equation=equation)
        cond = self.current_condition_name

-        r_norm = self.eta * torch.abs(residual) / (torch.max(torch.abs(residual))+1e-12)
+        r_norm = (
+            self.eta
+            * torch.abs(residual)
+            / (torch.max(torch.abs(residual)) + 1e-12)
+        )
        self.weights[cond] = (self.gamma * self.weights[cond] + r_norm).detach()

        loss_value = self._vectorial_loss(