From c746a478e4f3a15255e91393f4cedd7accda38b8 Mon Sep 17 00:00:00 2001 From: Roland Haas <rhaas@illinois.edu> Date: Wed, 15 May 2019 22:56:59 -0500 Subject: [PATCH] POWER: use sqrt rather than x**(1/2) this is faster and also safer if anyone uses python2 --- POWER/power.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/POWER/power.py b/POWER/power.py index 381bdef..4332219 100755 --- a/POWER/power.py +++ b/POWER/power.py @@ -461,11 +461,11 @@ if __name__ == "__main__": mp_psi4_vars[i][:, 2] *= radii[i] #Check for psi4 amplitude going to zero - cur_psi4_amp = (mp_psi4_vars[i][0, 1]**2 + mp_psi4_vars[i][0, 2]**2)**(1/2) + cur_psi4_amp = np.sqrt(mp_psi4_vars[i][0, 1]**2 + mp_psi4_vars[i][0, 2]**2) min_psi4_amp = cur_psi4_amp # TODO: use array notatino for this since it finds the minimum amplitude for j in range(0, len(mp_psi4_vars[i][:, 0])): - cur_psi4_amp = (mp_psi4_vars[i][j, 1]**2 + mp_psi4_vars[i][j, 2]**2)**(1/2) + cur_psi4_amp = np.sqrt(mp_psi4_vars[i][j, 1]**2 + mp_psi4_vars[i][j, 2]**2) if(cur_psi4_amp < min_psi4_amp): min_psi4_amp = cur_psi4_amp if(min_psi4_amp < np.finfo(float).eps and l >= 2): -- GitLab