Skip to content

Commit a109dbb

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 478de8c commit a109dbb

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
@@ -5984,7 +5984,9 @@ def remove_key(self, key):
59845984
ts._exception_blame = ts._exception = ts._traceback = None
59855985
self.task_metadata.pop(key, None)
59865986

5987-
def transition(self, key, finish, *args, **kwargs):
5987+
def transition(
5988+
self, key, finish, *args, status=None, thread=None, metadata=None, **kwargs
5989+
):
59885990
"""Transition a key from its current state to the finish state
59895991
59905992
Examples

0 commit comments

Comments
 (0)