Add join_threads() (#10086)

* Update __init__.py Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update __init__.py Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> * Update __init__.py Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
ultralytics · Nov 15, 2022 · 9dd40f0 · 9dd40f0
1 parent ea73386
commit 9dd40f0
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/utils/__init__.py b/utils/__init__.py
@@ -37,6 +37,16 @@ def wrapper(*args, **kwargs):
     return wrapper
 
 
+def join_threads(verbose=False):
+    # Join all daemon threads, i.e. atexit.register(lambda: join_threads())
+    main_thread = threading.current_thread()
+    for t in threading.enumerate():
+        if t is not main_thread:
+            if verbose:
+                print(f'Joining thread {t.name}')
+            t.join()
+
+
 def notebook_init(verbose=True):
     # Check system software and hardware
     print('Checking setup...')