Update pycbc_pygrb_plot_injs_results

Applying reweighted SNR cut to found injections also to the key storing reweighted SNR itself!
gwastro · Jan 24, 2025 · f3b926a · f3b926a
1 parent e693aa3
commit f3b926a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/bin/pygrb/pycbc_pygrb_plot_injs_results b/bin/pygrb/pycbc_pygrb_plot_injs_results
@@ -309,6 +309,7 @@ if opts.newsnr_threshold:
         found_inj[key] = found_inj[key][~rw_snr_cut]
     for ifo in ifos:
         found_inj[ifo+'/end_time'] = found_inj[ifo+'/end_time'][~rw_snr_cut]
+    found_inj['reweighted_snr'] = found_inj['reweighted_snr'][~rw_snr_cut]
     msg = f"After applying reweighted SNR cut at {opts.newsnr_threshold}: "
     msg += f"{len(found_inj[x_qty])} found injections and "
     msg += f"{len(missed_inj[x_qty])} missed injections"