Skip to content

Commit bbd2de4

Browse files
aseyboldttwiecki
authored andcommitted
Close progressbar in sample_ppc
1 parent 1778e91 commit bbd2de4

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

pymc3/sampling.py

+10-6
Original file line numberDiff line numberDiff line change
@@ -526,12 +526,16 @@ def sample_ppc(trace, samples=None, model=None, vars=None, size=None,
526526
else:
527527
indices = randint(0, len(trace), samples)
528528

529-
ppc = defaultdict(list)
530-
for idx in indices:
531-
param = trace[idx]
532-
for var in vars:
533-
ppc[var.name].append(var.distribution.random(point=param,
534-
size=size))
529+
try:
530+
ppc = defaultdict(list)
531+
for idx in indices:
532+
param = trace[idx]
533+
for var in vars:
534+
vals = var.distribution.random(point=param, size=size)
535+
ppc[var.name].append(vals)
536+
finally:
537+
if progressbar:
538+
indices.close()
535539

536540
return {k: np.asarray(v) for k, v in ppc.items()}
537541

0 commit comments

Comments
 (0)