Skip to content

Commit d245a01

Browse files
committed
Remove some refreshes that slow things down
1 parent ecabab5 commit d245a01

File tree

3 files changed

+2
-3
lines changed

3 files changed

+2
-3
lines changed

pymc/sampling/mcmc.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1083,7 +1083,7 @@ def _sample(
10831083
for it, diverging in enumerate(sampling_gen):
10841084
if it >= skip_first and diverging:
10851085
_pbar_data["divergences"] += 1
1086-
progress.update(task, refresh=True, advance=1)
1086+
progress.update(task)
10871087
progress.update(task, refresh=True, advance=1, completed=True)
10881088
except KeyboardInterrupt:
10891089
pass

pymc/sampling/parallel.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -476,7 +476,6 @@ def __iter__(self):
476476

477477
progress.update(
478478
task,
479-
refresh=True,
480479
completed=self._completed_draws,
481480
total=self._total_draws,
482481
description=self._desc.format(self),

pymc/sampling/population.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ def _sample_population(
104104
task = progress.add_task("[red]Sampling...", total=draws)
105105

106106
for _ in sampling:
107-
progress.update(task, advance=1, refresh=True)
107+
progress.update(task)
108108

109109
return
110110

0 commit comments

Comments
 (0)