Merge pull request #793 from CoffeaTeam/use_uproot_chunking

Use uproot chunking for uproot.dask, repin uproot
scikit-hep · Apr 14, 2023 · b9673d8 · b9673d8
2 parents 90e4930 + 43e5cd8
commit b9673d8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/coffea/nanoevents/factory.py b/coffea/nanoevents/factory.py
@@ -229,6 +229,7 @@ def from_root(
         treepath="/Events",
         entry_start=None,
         entry_stop=None,
+        chunks_per_file=1,
         runtime_cache=None,
         persistent_cache=None,
         schemaclass=NanoAODSchema,
@@ -316,6 +317,7 @@ def from_root(
                     open_files=False,
                     ak_add_doc=True,
                     filter_branch=_remove_not_interpretable,
+                    steps_per_file=chunks_per_file,
                 )
             else:
                 opener = partial(
@@ -325,6 +327,7 @@ def from_root(
                     open_files=False,
                     ak_add_doc=True,
                     filter_branch=_remove_not_interpretable,
+                    steps_per_file=chunks_per_file,
                 )
             return cls(map_schema, opener, None, cache=None, is_dask=True)
         elif permit_dask and not schemaclass.__dask_capable__:

diff --git a/setup.py b/setup.py
@@ -55,7 +55,7 @@ def get_description():
 
 INSTALL_REQUIRES = [
     "awkward>=2.1.3",
-    "uproot>=5.0.5",
+    "uproot>=5.0.7",
     "dask[array]>=2022.12.1",
     "dask-awkward>=2023.4.1",
     "dask-histogram>=2023.4.1",