From 43e5cd84454e7a0371f6ada09d698311a5747123 Mon Sep 17 00:00:00 2001 From: Lindsey Gray Date: Fri, 14 Apr 2023 11:26:05 -0500 Subject: [PATCH] use uproot chunking for uproot.dask, repin uproot --- coffea/nanoevents/factory.py | 3 +++ setup.py | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/coffea/nanoevents/factory.py b/coffea/nanoevents/factory.py index 8b355540f..0793960b8 100644 --- 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 index 824cedac9..2465f0a73 100644 --- 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",