diff --git a/bilby/gw/conversion.py b/bilby/gw/conversion.py index 4be5634eb2977fac17db7b7c22fe30f2d6591916..85bdf115a0091da5e01e370374011e26ab04953b 100644 --- a/bilby/gw/conversion.py +++ b/bilby/gw/conversion.py @@ -2261,8 +2261,8 @@ def compute_snrs(sample, likelihood, npool=1): snr_update = new_sample[ii].snrs_as_sample for key, val in snr_update.items(): snr_updates[f"{ifo.name}_{key}"].append(val) - for k, v in snr_updates.items(): - sample[k] = v + for k, v in snr_updates.items(): + sample[k] = v else: logger.debug('Not computing SNRs.') diff --git a/test/gw/conversion_test.py b/test/gw/conversion_test.py index aef17c822e741306146004725f6f827e551c6ea6..8ce40e3b887135202a10beb5e770efd3b754e1af 100644 --- a/test/gw/conversion_test.py +++ b/test/gw/conversion_test.py @@ -475,7 +475,7 @@ class TestGenerateAllParameters(unittest.TestCase): def test_generate_bbh_parameters_with_likelihood(self): priors = bilby.gw.prior.BBHPriorDict() priors["geocent_time"] = bilby.core.prior.Uniform(0.4, 0.6) - ifos = bilby.gw.detector.InterferometerList(["H1"]) + ifos = bilby.gw.detector.InterferometerList(["H1", "L1"]) ifos.set_strain_data_from_power_spectral_densities(duration=1, sampling_frequency=256) wfg = bilby.gw.waveform_generator.WaveformGenerator( frequency_domain_source_model=bilby.gw.source.lal_binary_black_hole @@ -501,6 +501,8 @@ class TestGenerateAllParameters(unittest.TestCase): "phase", "H1_optimal_snr", "H1_matched_filter_snr", + "L1_optimal_snr", + "L1_matched_filter_snr", "ra", "dec", ]