Skip to content

Commit c28de04

Browse files
committed
Filter some arguments out in transition
To avoid passing unneeded arguments to `transition_*` functions, catch them in `transition` with keyword arguments.
1 parent 2c2c4c3 commit c28de04

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

distributed/scheduler.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5974,7 +5974,9 @@ def remove_key(self, key):
59745974
ts._exception_blame = ts._exception = ts._traceback = None
59755975
self.task_metadata.pop(key, None)
59765976

5977-
def transition(self, key, finish, *args, **kwargs):
5977+
def transition(
5978+
self, key, finish, *args, status=None, thread=None, metadata=None, **kwargs
5979+
):
59785980
"""Transition a key from its current state to the finish state
59795981
59805982
Examples

0 commit comments

Comments
 (0)