Use a label formatter for prettier histograms.

[python_utils.git] / executors.py

diff --git a/executors.py b/executors.py
index 5b77a42dc3d29ca6f42673a369e23f0962343c62..28507b0e8a547e9d1edc615daa5de7945edb457d 100644 (file)
--- a/executors.py
+++ b/executors.py
@@ -152,12 +152,13 @@ class ThreadExecutor(BaseExecutor):
         )
         result.add_done_callback(lambda _: self.histogram.add_item(time.time() - start))
         result.add_done_callback(lambda _: self.adjust_task_count(-1))
+        return result
 
     @overrides
     def shutdown(self, wait=True) -> None:
         if not self.already_shutdown:
             logger.debug(f'Shutting down threadpool executor {self.title}')
-            print(self.histogram)
+            print(self.histogram.__repr__(label_formatter='%d'))
             self._thread_pool_executor.shutdown(wait)
             self.already_shutdown = True
 
@@ -200,7 +201,7 @@ class ProcessExecutor(BaseExecutor):
         if not self.already_shutdown:
             logger.debug(f'Shutting down processpool executor {self.title}')
             self._process_executor.shutdown(wait)
-            print(self.histogram)
+            print(self.histogram.__repr__(label_formatter='%d'))
             self.already_shutdown = True
 
     def __getstate__(self):
@@ -1112,7 +1113,7 @@ class RemoteExecutor(BaseExecutor):
             self.heartbeat_stop_event.set()
             self.heartbeat_thread.join()
             self._helper_executor.shutdown(wait)
-            print(self.histogram)
+            print(self.histogram.__repr__(label_formatter='%d'))
             self.already_shutdown = True