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

fix(app): create reference to events task to prevent accidental GC

Browse Source 
This wasn't a problem, but it's advised in the official docs so I've done it.
This commit is contained in:
psychedelicious 2024-08-04 16:26:43 +10:00
parent 29325a7214
commit f6b8970bd1
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/app/services/events/events_fastapievents.py
View File

@ -13,7 +13,13 @@ class FastAPIEventService(EventServiceBase):
self._queue = asyncio.Queue[EventBase | None]()
self._stop_event = threading.Event()
self._loop = loop
self._loop.create_task(self._dispatch_from_queue(stop_event=self._stop_event))
# We need to store a reference to the task so it doesn't get GC'd
# See: https://docs.python.org/3/library/asyncio-task.html#creating-tasks
self._background_tasks: set[asyncio.Task[None]] = set()
task = self._loop.create_task(self._dispatch_from_queue(stop_event=self._stop_event))
self._background_tasks.add(task)
task.add_done_callback(self._background_tasks.remove)
super().__init__()