Skip to content

Commit e3f008a

Browse files
jucortwiecki
authored andcommitted
Fix progress bar for parallel sampling
1 parent 3a59c16 commit e3f008a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

pymc/smc/sampling.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
import numpy as np
2626

2727
from arviz import InferenceData
28-
from fastprogress.fastprogress import progress_bar
28+
from fastprogress.fastprogress import force_console_behavior, progress_bar
2929

3030
import pymc
3131

@@ -376,6 +376,8 @@ def _sample_smc_int(
376376

377377

378378
def run_chains_parallel(chains, progressbar, to_run, params, random_seed, kernel_kwargs, cores):
379+
# fastprogress HTML progress bar does not support multiprocessing
380+
_, progress_bar = force_console_behavior()
379381
pbar = progress_bar((), total=100, display=progressbar)
380382
pbar.update(0)
381383
pbars = [pbar] + [None] * (chains - 1)

0 commit comments

Comments
 (0)