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",