Skip to content

Commit ffff317

Browse files
authored
feat: set divisions to unknown if we expect a read error report in uproot.dask (#1543)
set divisions to None if we expect a report
1 parent 8a5f52d commit ffff317

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

src/uproot/_dask.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1731,12 +1731,15 @@ def real_filter_branch(branch):
17311731
interpretation_executor=interpretation_executor,
17321732
)
17331733

1734-
return dask_awkward.from_map(
1734+
out = dask_awkward.from_map(
17351735
fn,
17361736
partition_args,
17371737
divisions=tuple(divisions),
17381738
label="from-uproot",
17391739
)
1740+
if allow_read_errors_with_report:
1741+
out[0]._divisions = tuple(None for i in out[0]._divisions)
1742+
return out
17401743

17411744

17421745
def _get_dak_array_delay_open(
@@ -1838,12 +1841,15 @@ def _get_dak_array_delay_open(
18381841
interpretation_executor=interpretation_executor,
18391842
)
18401843

1841-
return dask_awkward.from_map(
1844+
out = dask_awkward.from_map(
18421845
fn,
18431846
partition_args,
18441847
divisions=None if divisions is None else tuple(divisions),
18451848
label="from-uproot",
18461849
)
1850+
if allow_read_errors_with_report:
1851+
out[0]._divisions = tuple(None for i in out[0]._divisions)
1852+
return out
18471853

18481854

18491855
class Accessed(NamedTuple):

0 commit comments

Comments
 (0)