fix(task-manager): force a flushing queues atexit (#1006)

glesperance · web-flow · commit dd254929d64b · 2024-11-26T18:08:54.000+01:00
diff --git a/langfuse/_task_manager/task_manager.py b/langfuse/_task_manager/task_manager.py
@@ -81,7 +81,7 @@ def __init__(
         self.init_resources()
 
         # cleans up when the python interpreter closes
-        atexit.register(self.join)
+        atexit.register(self.shutdown)
 
     def init_resources(self):
         for i in range(self._threads):
