flush atpbar in ThreadingHub

alphatwirl · Jun 26, 2019 · 228ab03 · 228ab03
1 parent 984e1f8
commit 228ab03
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/mantichora/hubthreading.py b/mantichora/hubthreading.py
@@ -4,6 +4,8 @@
 
 from operator import itemgetter
 
+import atpbar
+
 ##__________________________________________________________________||
 class ThreadingHub(object):
     """A hub for Threading
@@ -137,6 +139,8 @@ def receive(self):
         # sort in the order of task_idx
         messages = sorted(messages, key=itemgetter(0))
 
+        atpbar.flush()
+
         return messages
 
     def _receive_finished(self):
@@ -171,6 +175,8 @@ def close(self):
             self.task_queue.join()
             self.workers = [ ]
 
+        atpbar.flush()
+
 ##__________________________________________________________________||
 import logging
 class Worker(threading.Thread):