fallback to tree.num_entries for now

scikit-hep · Oct 23, 2024 · fa12703 · fa12703
1 parent 600629e
commit fa12703
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/coffea/nanoevents/factory.py b/src/coffea/nanoevents/factory.py
@@ -374,12 +374,10 @@ def from_root(
 
         if entry_start is None or entry_start < 0:
             entry_start = 0
-        if entry_stop is None or entry_stop > tree.num_entries:
-            if treepath is not None:
-                # if treepath exists, attempt the faster uproot.num_entries method
-                entry_stop = next(uproot.num_entries(f"{file}:{treepath}"))[-1]
-            else:
-                entry_stop = tree.num_entries
+        # should use uproot.num_entries instead...
+        nentries = tree.num_entries
+        if entry_stop is None or entry_stop > nentries:
+                entry_stop = nentries
 
         partition_key = (
             str(tree.file.uuid),