make use of fstatVar if/when available

OstapHEP · Apr 25, 2024 · 3037bb8 · 3037bb8
1 parent 6844564
commit 3037bb8
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/ostap/trees/trees.py b/ostap/trees/trees.py
@@ -1298,11 +1298,12 @@ def _rt_table_0_ ( tree ,
 
         bbs.append ( b ) 
 
-    if hasattr ( tree , 'pstatVar' ) : bbstats = tree.pstatVar ( bbs , cuts , *args )
-    else                             : bbstats = tree. statVar ( bbs , cuts , *args )
+    if   hasattr ( tree , 'fstatVar' ) : bbstats = tree.fstatVar ( bbs , cuts , *args )
+    elif hasattr ( tree , 'pstatVar' ) : bbstats = tree.pstatVar ( bbs , cuts , *args )
+    else                               : bbstats = tree. statVar ( bbs , cuts , *args )
 
     from ostap.stats.counters import WSE, SE  
-    if   isinstance ( bbstats , ( WSE , SE ) )  : bbstats = { bbs[0] : bbstats } 
+    if   isinstance ( bbstats , ( WSE , SE ) )  : bbstats = { bbs [ 0 ] : bbstats } 
 
     for b in brs :