protect against exception when non-traversed list is different length

dask-contrib · Dec 16, 2024 · d0f41d1 · d0f41d1
1 parent 78428ab
commit d0f41d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/dask_awkward/lib/core.py b/src/dask_awkward/lib/core.py
@@ -2163,7 +2163,7 @@ def map_partitions(
 
     if len(kwargs) == 0:
         non_traversed_deps, _ = unpack_collections(*args, traverse=False)
-        if all(
+        if len(flat_deps) == len(non_traversed_deps) and all(
             id(traversed_dep) == id(non_traversed_dep)
             for traversed_dep, non_traversed_dep in zip(flat_deps, non_traversed_deps)
         ):