From 1f6ea277ad339a8f7abf526ae392dfde5f3603d9 Mon Sep 17 00:00:00 2001 From: ndem0 Date: Thu, 10 Oct 2024 16:46:27 +0000 Subject: [PATCH] :art: Format Python code with psf/black --- pina/solvers/pinns/rbapinn.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/pina/solvers/pinns/rbapinn.py b/pina/solvers/pinns/rbapinn.py index b651b0b..fd551ac 100644 --- 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(