Skip to content

Commit d6de023

Browse files
Fix cancelled status (#110)
* fix cancelled status * fix status cancelled * remove print
1 parent 2ae1aa6 commit d6de023

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

cads_broker/dispatcher.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -364,15 +364,15 @@ def on_future_done(self, future: distributed.Future) -> None:
364364
logger.info("worker killed: re-queueing", job_id=future.key)
365365
db.requeue_request(request_uid=future.key, session=session)
366366
self.queue.add(future.key, request)
367-
elif future.status != "cancelled":
367+
else:
368368
request = db.set_request_status(
369369
future.key,
370370
job_status,
371371
error_message=error_message,
372372
error_reason=error_reason,
373373
session=session,
374374
)
375-
else:
375+
elif future.status != "cancelled":
376376
# if the dask status is unknown, re-queue it
377377
request = db.set_request_status(
378378
future.key,
@@ -386,6 +386,9 @@ def on_future_done(self, future: distributed.Future) -> None:
386386
job_status={future.status},
387387
job_id=request.request_uid,
388388
)
389+
else:
390+
# if the dask status is cancelled, the qos has already been reset by sync_database
391+
return
389392
self.futures.pop(future.key, None)
390393
self.qos.notify_end_of_request(
391394
request, session, scheduler=self.internal_scheduler

0 commit comments

Comments
 (0)