consolidate progress bar function

Author: williambdean

pymc/util.py

@@ -859,15 +859,7 @@ def _initialize_tasks(self):
 
     def compute_draw_speed(self, chain_idx, draws):
         elapsed = self._progress.tasks[chain_idx].elapsed
-        speed = draws / max(elapsed, 1e-6)
-
-        if speed > 1 or speed == 0:
-            unit = "draws/s"
-        else:
-            unit = "s/draws"
-            speed = 1 / speed
-
-        return speed, unit
+        return compute_draw_speed(elapsed, draws)
 
     def update(self, chain_idx, is_last, draw, tuning, stats):
         if not self._show_progress: