mapfilter: fix args passing

pfackeldey · Nov 19, 2024 · 1026471 · 1026471
1 parent 4213659
commit 1026471
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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