adapt to scikit-hep/uproot5#1050

scikit-hep · Nov 29, 2023 · d664317 · d664317
1 parent 23a8f32
commit d664317
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/coffea/nanoevents/factory.py b/src/coffea/nanoevents/factory.py
@@ -661,7 +661,12 @@ def events(self):
         """Build events"""
         if self._is_dask:
             events = self._mapping(form_mapping=self._schema)
+            report = None
+            if isinstance(events, tuple):
+                events, report = events
             events.behavior["__original_array__"] = lambda: events
+            if report is not None:
+                return events, report
             return events
 
         events = self._events()