Skip to content

Commit a164f72

Browse files
authored
Small update of VI and API doc (#3795)
1 parent 80237a1 commit a164f72

File tree

2 files changed

+396
-170
lines changed

2 files changed

+396
-170
lines changed

docs/source/notebooks/variational_api_quickstart.ipynb

Lines changed: 395 additions & 169 deletions
Large diffs are not rendered by default.

pymc3/variational/inference.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -281,7 +281,7 @@ def refine(self, n, progressbar=True):
281281
raise TypeError("Need to call `.fit` first")
282282
i, step, callbacks, score = self.state
283283
if progressbar:
284-
progress = progress_bar(n, display=progressbar)
284+
progress = progress_bar(range(n), display=progressbar)
285285
else:
286286
progress = range(n) # This is a guess at what progress_bar(n) does.
287287
if score:

0 commit comments

Comments
 (0)