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
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.