Merge pull request #887 from iasonkrom/pass_uproot_options_to_uproot_…

…dask feat: also pass `uproot_options` to `uproot.dask` in `from_root()`
scikit-hep · Sep 6, 2023 · 583bc1a · 583bc1a
2 parents 2f5af18 + f49febd
commit 583bc1a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/coffea/nanoevents/factory.py b/src/coffea/nanoevents/factory.py
@@ -232,7 +232,7 @@ def from_root(
         treepath="/Events",
         entry_start=None,
         entry_stop=None,
-        chunks_per_file=1,
+        chunks_per_file=uproot._util.unset,
         runtime_cache=None,
         persistent_cache=None,
         schemaclass=NanoAODSchema,
@@ -268,7 +268,7 @@ def from_root(
             metadata : dict, optional
                 Arbitrary metadata to add to the `base.NanoEvents` object
             uproot_options : dict, optional
-                Any options to pass to ``uproot.open``
+                Any options to pass to ``uproot.open`` or ``uproot.dask``
             access_log : list, optional
                 Pass a list instance to record which branches were lazily accessed by this instance
             use_ak_forth:
@@ -326,6 +326,7 @@ def from_root(
                     ak_add_doc=True,
                     filter_branch=_remove_not_interpretable,
                     steps_per_file=chunks_per_file,
+                    **uproot_options,
                 )
             else:
                 opener = partial(
@@ -336,6 +337,7 @@ def from_root(
                     ak_add_doc=True,
                     filter_branch=_remove_not_interpretable,
                     steps_per_file=chunks_per_file,
+                    **uproot_options,
                 )
             return cls(map_schema, opener, None, cache=None, is_dask=True)
         elif permit_dask and not schemaclass.__dask_capable__: