fix process state prints

gwpy · Sep 9, 2024 · a85b0ea · a85b0ea
1 parent 83ba19f
commit a85b0ea
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/gwsumm/tabs/data.py b/gwsumm/tabs/data.py
@@ -338,8 +338,9 @@ def finalize_states(self, config=ConfigParser(), segdb_error='raise',
         for state in self.states:
             state.fetch(config=config, segdb_error=segdb_error, **kwargs)
         # finilize statebar
-        if self.statebar:
-            self.statebar.fetch(config=config, segdb_error=segdb_error, **kwargs)
+        if self.statebar is not None:
+            self.statebar.fetch(config=config, segdb_error=segdb_error,
+                                **kwargs)
 
 
     def process(self, config=ConfigParser(), nproc=1, **stateargs):
@@ -365,12 +366,16 @@ def process(self, config=ConfigParser(), nproc=1, **stateargs):
             nproc=nproc, nds=stateargs.get('nds', None))
         vprint(f"States finalised [{len(self.states) + len([self.statebar])}"
                " total]\n")
-        for state in self.states + [self.statebar]:
+        for state in self.states:
             vprint(f"    {state.name}: {len(state.active)} segments"
-                   " | {abs(state.active)} seconds")
+                   f" | {abs(state.active)} seconds")
             if state is self.defaultstate:
                 vprint(" [DEFAULT]")
             vprint('\n')
+        if self.statebar is not None:
+            vprint(f"    {self.statebar.name}: {len(self.statebar.active)}"
+                   f" segments | {abs(state.active)} seconds")
+            vprint('\n')
 
         # pre-process requests for 'all-data' plots
         all_data = any([(p.all_data & p.new) for p in self.plots])