From 37927dea78217e9b9374f2f82a4aa5fbd5a5f841 Mon Sep 17 00:00:00 2001 From: Peter Pang Date: Sun, 28 Jan 2024 16:43:32 +0100 Subject: [PATCH] Renaming psd-related variables to avoid confusion --- src/jimgw/single_event/detector.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/jimgw/single_event/detector.py b/src/jimgw/single_event/detector.py index c8f813ab..4cea21af 100644 --- a/src/jimgw/single_event/detector.py +++ b/src/jimgw/single_event/detector.py @@ -15,7 +15,7 @@ DEG_TO_RAD = jnp.pi / 180 # TODO: Need to expand this list. Currently it is only O3. -psd_file_dict = { +asd_file_dict = { "H1": "https://dcc.ligo.org/public/0169/P2000251/001/O3-H1-C01_CLEAN_SUB60HZ-1251752040.0_sensitivity_strain_asd.txt", "L1": "https://dcc.ligo.org/public/0169/P2000251/001/O3-L1-C01_CLEAN_SUB60HZ-1240573680.0_sensitivity_strain_asd.txt", "V1": "https://dcc.ligo.org/public/0169/P2000251/001/O3-V1_sensitivity_strain_asd.txt", @@ -412,13 +412,13 @@ def load_psd( ) -> Float[Array, " n_sample"]: if psd_file == "": print("Grabbing GWTC-2 PSD for " + self.name) - url = psd_file_dict[self.name] + url = asd_file_dict[self.name] data = requests.get(url) open(self.name + ".txt", "wb").write(data.content) f, asd_vals = np.loadtxt(self.name + ".txt", unpack=True) + psd_vals = asd_vals**2 else: - f, asd_vals = np.loadtxt(psd_file, unpack=True) - psd_vals = asd_vals**2 + f, psd_vals = np.loadtxt(psd_file, unpack=True) psd = interp1d(f, psd_vals, fill_value=(psd_vals[0], psd_vals[-1]))(freqs) # type: ignore psd = jnp.array(psd)