Add refresh call to progress bar in apply_function_over_dataset (#7343)

fonnesbeck · web-flow · commit 09ea8c83d7cb · 2024-06-02T21:30:05.000-04:00
diff --git a/pymc/backends/arviz.py b/pymc/backends/arviz.py
@@ -660,6 +660,7 @@ def apply_function_over_dataset(
                 out_dict.insert(var_name, val, idx)
 
             progress.advance(task)
+        progress.update(task, refresh=True, completed=n_pts)
 
     out_trace = out_dict.trace_dict
     for key, val in out_trace.items():
