adjust test for PR dask-contrib#558

pfackeldey · Dec 17, 2024 · 39eb8f3 · 39eb8f3
1 parent 5c4eebe
commit 39eb8f3
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -986,6 +986,8 @@ def test_map_partitions_deterministic_token():
     dask_arr = dak.from_awkward(ak.Array([1]), npartitions=1)
 
     def f(x):
-        return x + 1
+        return x[0] + 1
 
-    assert map_partitions(f, dask_arr).name == map_partitions(f, dask_arr).name
+    assert (
+        map_partitions(f, {0: dask_arr}).name == map_partitions(f, {0: dask_arr}).name
+    )