Skip to content

Commit 945fd71

Browse files
aseyboldttwiecki
authored andcommitted
Fix nuts initialization docstring (#2492)
1 parent ed2aa49 commit 945fd71

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pymc3/sampling.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def sample(draws=500, step=None, init='auto', n_init=200000, start=None,
123123
Initialization method to use for auto-assigned NUTS samplers.
124124
125125
* auto : Choose a default initialization method automatically.
126-
Currently, this is `'unif+adapt_diag'`, but this can change in
126+
Currently, this is `'jitter+adapt_diag'`, but this can change in
127127
the future. If you depend on the exact behaviour, choose an
128128
initialization method explicitly.
129129
* adapt_diag : Start with a identity mass matrix and then adapt
@@ -311,8 +311,8 @@ def _check_start_shape(model, start):
311311
if var_shape:
312312
e += "\nExpected shape {} for var " \
313313
"'{}', got scalar {}".format(
314-
tuple(var_shape), var.name, start[var.name]
315-
)
314+
tuple(var_shape), var.name, start[var.name]
315+
)
316316

317317
if e != '':
318318
raise ValueError("Bad shape for start argument:{}".format(e))
@@ -699,7 +699,7 @@ def init_nuts(init='auto', njobs=1, n_init=500000, model=None,
699699
Initialization method to use.
700700
701701
* auto : Choose a default initialization method automatically.
702-
Currently, this is `'unif+adapt_diag'`, but this can change in
702+
Currently, this is `'jitter+adapt_diag'`, but this can change in
703703
the future. If you depend on the exact behaviour, choose an
704704
initialization method explicitly.
705705
* adapt_diag : Start with a identity mass matrix and then adapt

0 commit comments

Comments
 (0)