Add update method to CustomProgress

pymc-devs · May 1, 2024 · 4fd6f32 · 4fd6f32
1 parent d2bb020
commit 4fd6f32
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/pymc/sampling/parallel.py b/pymc/sampling/parallel.py
@@ -479,9 +479,7 @@ def __iter__(self):
                     self._active.remove(proc)
                     self._finished.append(proc)
                     self._make_active()
-
-                    if self._progress.is_enabled:
-                        progress.update(task, description=self._desc.format(self), refresh=True)
+                    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.

diff --git a/pymc/util.py b/pymc/util.py
@@ -531,7 +531,7 @@ class CustomProgress(Progress):
     """
 
     def __init__(self, *args, **kwargs):
-        self.is_enabled = not kwargs.get("disable", None) is True
+        self.is_enabled = kwargs.get("disable", None) is not True
         if self.is_enabled:
             super().__init__(*args, **kwargs)