Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Run ruff

Browse Source
This commit is contained in:
brandonrising 2024-03-28 14:37:43 -04:00 committed by psychedelicious
parent 98cc9b963c
commit 43bcedee10
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
invokeai/app/services/session_processor/session_processor_default.py
View File

@ -70,13 +70,18 @@ class DefaultSessionProcessor(SessionProcessorBase):
async def _on_queue_event(self, event: FastAPIEvent) -> None: async def _on_queue_event(self, event: FastAPIEvent) -> None:
event_name = event[1]["event"] event_name = event[1]["event"]
if (
if event_name == "session_canceled" and self._queue_item\ event_name == "session_canceled"
and self._queue_item.item_id == event[1]["data"]["queue_item_id"]: and self._queue_item
and self._queue_item.item_id == event[1]["data"]["queue_item_id"]
):
self._cancel_event.set() self._cancel_event.set()
self._poll_now() self._poll_now()
elif event_name == "queue_cleared" and self._queue_item\ elif (
and self._queue_item.queue_id == event[1]["data"]["queue_id"]: event_name == "queue_cleared"
and self._queue_item
and self._queue_item.queue_id == event[1]["data"]["queue_id"]
):
self._cancel_event.set() self._cancel_event.set()
self._poll_now() self._poll_now()
elif event_name == "batch_enqueued": elif event_name == "batch_enqueued":