Improve logic for getting ifo frm trigs

gwastro · Jul 22, 2024 · d82cbca · d82cbca
1 parent b88025c
commit d82cbca
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/pycbc/events/stat.py b/pycbc/events/stat.py
@@ -2169,10 +2169,9 @@ def find_dq_noise_rate(self, trigs):
         try:
             ifo = trigs.ifo
         except AttributeError:
-            ifo = trigs['ifo']
-            assert len(numpy.unique(ifo)) == 1
             # Should be exactly one ifo provided
-            ifo = ifo[0]
+            assert len(numpy.unique(trigs['ifo'])) == 1
+            ifo = trigs['ifo'][0]
 
         dq_state = trigs['dq_state']
         dq_val = numpy.ones(len(dq_state))
@@ -2230,13 +2229,11 @@ def single(self, trigs):
         # make sure every trig has a dq state
 
         try:
-            # works in offline
             ifo = trigs.ifo
         except AttributeError:
-            # works in low-latency
-            ifo = trigs['ifo'][0]
             # Should be exactly one ifo provided
             assert len(numpy.unique(trigs['ifo'])) == 1
+            ifo = trigs['ifo'][0]
 
         singles = ExpFitFgBgNormStatistic.single(self, trigs)