From 3a8ee10af6fbd42a6dc21b6ddac56798e6e519e6 Mon Sep 17 00:00:00 2001 From: Shichao Wu Date: Tue, 4 Jul 2023 22:09:51 +0200 Subject: [PATCH] Update analytical_space.py --- pycbc/psd/analytical_space.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/pycbc/psd/analytical_space.py b/pycbc/psd/analytical_space.py index 8eca298be75..3654c82fadb 100644 --- a/pycbc/psd/analytical_space.py +++ b/pycbc/psd/analytical_space.py @@ -369,19 +369,20 @@ def averaged_lisa_fplus_sq_approx(f, len_arm=2.5e9): ----- Pease see Eq.(36) in for more details. """ - from os import getcwd, path + from os import path import requests from scipy.interpolate import interp1d + import pycbc.psd if len_arm != 2.5e9: raise Exception("Currently only support 'len_arm=2.5e9'.") - cwd = getcwd() - if path.exists(cwd+"/AvFXp2_Raw.npy") is False: + module_path = pycbc.psd.__path__[0] + if path.exists(module_path+"/AvFXp2_Raw.npy") is False: url = "https://zenodo.org/record/7497853/files/AvFXp2_Raw.npy" response = requests.get(url, verify=False) - with open(cwd+"/AvFXp2_Raw.npy", 'wb') as f: - f.write(response.content) - freqs, fp_sq = np.load(cwd+"/AvFXp2_Raw.npy") + with open(module_path+"/AvFXp2_Raw.npy", 'wb') as file: + file.write(response.content) + freqs, fp_sq = np.load(module_path+"/AvFXp2_Raw.npy") # Padding the end. freqs = np.append(freqs, 2) fp_sq = np.append(fp_sq, 0.0012712348970728724)