dask-contrib / dask-awkward

Native Dask collection for awkward arrays, and the library to use it.
https://dask-awkward.readthedocs.io
BSD 3-Clause "New" or "Revised" License
61 stars 19 forks source link

Left- vs right-multiplying array by Dask scalar #528

Open alexander-held opened 4 months ago

alexander-held commented 4 months ago

The following raises

NotImplementedError: no overload for ak.to_layout with arguments (dask.awkward<from-awkward, npartitions=1>,)

when attempting to multiply the array by a scalar from the left:

import awkward as ak
import dask_awkward as dak

arr = dak.from_awkward(ak.Array([1,3,4]), npartitions=1)
dak.sum(arr, axis=-1) * arr

Doing the same thing from the right works:

arr * dak.sum(arr, axis=-1)

I'm using awkward 2.6.6 and dask_awkward 2024.6.0.