Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
put only histref in the partial
Browse files Browse the repository at this point in the history
douglasdavis committed Dec 18, 2024

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
1 parent bf85767 commit 7f77ddf
Showing 1 changed file with 10 additions and 2 deletions.
12 changes: 10 additions & 2 deletions src/dask_histogram/core.py
Original file line number Diff line number Diff line change
@@ -1026,9 +1026,17 @@ def _partitioned_histogram(
# Single awkward array object.
if len(data) == 1 and data_is_dak:
from dask_awkward.lib.core import partitionwise_layer as dak_pwl
f = partial(_blocked_dak, histref=histref)

f = partial(_blocked_dak, weights=weights, sample=sample, histref=histref)
g = dak_pwl(f, name, data[0])
x = data[0]
if weights is not None and sample is not None:
g = dak_pwl(f, name, x, weights, sample)
elif weights is not None and sample is None:
g = dak_pwl(f, name, x, weights, None)
elif weights is None and sample is not None:
g = dak_pwl(f, name, x, None, sample)
else:
g = dak_pwl(f, name, x, None, None)

# Single object, not a dataframe
elif len(data) == 1 and not data_is_df:

0 comments on commit 7f77ddf

Please sign in to comment.