Prevent _run_request_notifier() uncatched exceptions

rapidsai · Oct 5, 2023 · 7590ec0 · 7590ec0
1 parent 24d8a2d
commit 7590ec0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/ucxx/_lib_async/notifier_thread.py b/python/ucxx/_lib_async/notifier_thread.py
@@ -12,7 +12,7 @@
 
 
 async def _run_request_notifier(worker):
-    worker.run_request_notifier()
+    return worker.run_request_notifier()
 
 
 async def _notifier_coroutine(worker):
@@ -58,4 +58,7 @@ def _notifierThread(event_loop, worker, q):
         try:
             task.result(0.01)
         except TimeoutError:
+            task.cancel()
             logger.debug("Notifier Thread Result Timeout")
+        except Exception as e:
+            logger.debug(f"Notifier Thread Result Exception: {e}")