From 0562772d42fbb2f44453846512c589ca2625f62e Mon Sep 17 00:00:00 2001 From: Logan Bishop-Van Horn Date: Thu, 27 Jun 2024 16:04:29 -0700 Subject: [PATCH] Use xp.allclose() --- tdgl/solver/solver.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tdgl/solver/solver.py b/tdgl/solver/solver.py index 73b0544..3425c23 100644 --- a/tdgl/solver/solver.py +++ b/tdgl/solver/solver.py @@ -632,7 +632,7 @@ def update( (current_A_applied - prev_A_applied) / dt, self.normalized_directions, ) - if xp.any(xp.absolute(dA_dt) > 0): + if not xp.allclose(current_A_applied, self.current_A_applied): # Update the link exponents only if the applied vector potential # has actually changed. operators.set_link_exponents(current_A_applied) @@ -642,10 +642,10 @@ def update( self.current_A_applied = current_A_applied # Update the value of epsilon - epsilon = self.epsilon if self.dynamic_epsilon: - epsilon = self.epsilon = self.update_epsilon(time) + self.epsilon = self.update_epsilon(time) + epsilon = self.epsilon old_sq_psi = xp.absolute(psi) ** 2 screening_error = np.inf A_induced_vals = [A_induced]