Set `WorkerState.processing` w/`dict` in `clean`
Based on the documentation about `processing`, this should be a `dict`.
However here we are assigning it a `set`. Here we simply pass along the
values along with the keys.
@@ -386,7 +386,7 @@ class WorkerState:
nanny=self.nanny,
extra=self.extra,
)
- ws.processing = {ts.key for ts in self.processing}
+ ws.processing = {ts.key: cost for ts, cost in self.processing.items()}
return ws
def __repr__(self):
