From 1026471ea9246fb53295504644771299bff6045d Mon Sep 17 00:00:00 2001 From: pfackeldey Date: Tue, 19 Nov 2024 14:12:38 +0100 Subject: [PATCH] mapfilter: fix args passing --- src/dask_awkward/lib/decorator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/dask_awkward/lib/decorator.py b/src/dask_awkward/lib/decorator.py index d8608519..aa04d83f 100644 --- a/src/dask_awkward/lib/decorator.py +++ b/src/dask_awkward/lib/decorator.py @@ -323,12 +323,12 @@ def __call__(self, *args: tp.Any, **kwargs: tp.Any) -> tp.Any: hlg, meta, deps, name = _map_partitions_prepare( fn, - *args, + *arg_flat_deps_expanded, + *kwarg_flat_deps, label=self.label, token=self.token, meta=self.meta, output_divisions=None, - **kwargs, ) # check consistent partitioning