Remove recursion in task spec

dask · Oct 31, 2024 · ae61cd6 · ae61cd6
1 parent 9da5824
commit ae61cd6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/distributed/shuffle/_core.py b/distributed/shuffle/_core.py
@@ -564,7 +564,7 @@ def _mean_shard_size(shards: Iterable) -> int:
     return size // count if count else 0
 
 
-def p2p_barrier(id: ShuffleId, run_ids: list[int]) -> int:
+def p2p_barrier(id: ShuffleId, *run_ids: int) -> int:
     try:
         return get_worker_plugin().barrier(id, run_ids)
     except Reschedule as e: