diff --git a/pycbc/events/stat.py b/pycbc/events/stat.py index 185c377361d..a3ad84f0765 100644 --- a/pycbc/events/stat.py +++ b/pycbc/events/stat.py @@ -2166,11 +2166,6 @@ def setup_segments(self, key): def find_dq_noise_rate(self, trigs): """Get dq values for a specific ifo and dq states""" - try: - tnum = trigs.template_num - except AttributeError: - tnum = trigs['template_id'] - try: ifo = trigs.ifo except AttributeError: @@ -2180,8 +2175,9 @@ def find_dq_noise_rate(self, trigs): ifo = ifo[0] dq_state = trigs['dq_state'] - dq_val = numpy.zeros(len(dq_state)) + dq_val = numpy.ones(len(dq_state)) + tnum = self.curr_tnum if ifo in self.dq_rates_by_state: for (i, st) in enumerate(dq_state): if isinstance(tnum, numpy.ndarray):