From 39eb8f3fd56fa2f76fc08102f412d28157ab5385 Mon Sep 17 00:00:00 2001
From: pfackeldey <fackeldey.peter@gmail.com>
Date: Tue, 17 Dec 2024 12:54:25 -0500
Subject: [PATCH] adjust test for PR #558

---
 tests/test_core.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/test_core.py b/tests/test_core.py
index cb99890c..627b04c3 100644
--- 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
+    )