Replace Progress with CustomProgress

Author: tomicapretto

pymc/sampling/parallel.py

@@ -487,7 +487,9 @@ def __iter__(self):
                     self._active.remove(proc)
                     self._finished.append(proc)
                     self._make_active()
-                    progress.update(task, description=self._desc.format(self), refresh=True)
+
+                    if self._progress.is_enabled:
+                        progress.update(task, description=self._desc.format(self), refresh=True)
 
                 # We could also yield proc.shared_point_view directly,
                 # and only call proc.write_next() after the yield returns.

pymc/util.py

@@ -531,7 +531,7 @@ class CustomProgress(Progress):
     """
 
     def __init__(self, *args, **kwargs):
-        self.is_enabled = kwargs.get("disable", None) is not True
+        self.is_enabled = not kwargs.get("disable", None) is True
         if self.is_enabled:
             super().__init__(*args, **kwargs)