Get number of sets right for bar updating

gwastro · Jul 24, 2023 · 232ff0d · 232ff0d
1 parent 61c17b4
commit 232ff0d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bin/pygrb/pycbc_grb_trig_cluster b/bin/pygrb/pycbc_grb_trig_cluster
@@ -72,7 +72,8 @@ def slice_hdf5(inputfile, outfile, include, verbose=False):
             ) for ifo in ifos
         }
 
-        nsets = sum(isinstance(item, h5py.Dataset) for
+        nsets = sum(isinstance(item, h5py.Dataset)
+                    or isinstance(item, h5py.Group) for
                     group in h5in.values() for
                     item in group.values())
         msg = "Slicing {} network events into new file".format(nevents)