BUG: Fix crash when exiting the daemon thread (#116)

xorbitsai · Nov 29, 2024 · 0f28fe7 · 0f28fe7
1 parent b1fd262
commit 0f28fe7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/xoscar/backends/core.py b/python/xoscar/backends/core.py
@@ -16,6 +16,7 @@
 from __future__ import annotations
 
 import asyncio
+import atexit
 import copy
 import logging
 import threading
@@ -216,6 +217,7 @@ class _RefHolder:
     _close_loop = asyncio.new_event_loop()
     _close_thread = threading.Thread(target=_close_loop.run_forever, daemon=True)
     _close_thread.start()
+    atexit.register(_close_loop.call_soon_threadsafe, _close_loop.stop)
 
     def __init__(self):
         self._thread_local = threading.local()