Skip to content

Commit 97a7a00

Browse files
aloctavodiaricardoV94
authored andcommitted
Reduce number of minimum draw in warning message
1 parent 7b38a48 commit 97a7a00

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

pymc/sampling/mcmc.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -665,8 +665,11 @@ def sample(
665665
if draws == 0:
666666
msg = "Tuning was enabled throughout the whole trace."
667667
_log.warning(msg)
668-
elif draws < 500:
669-
msg = "Only %s samples in chain." % draws
668+
elif draws < 100:
669+
msg = (
670+
"Only %s samples per chain. Reliable r-hat and ESS diagnostics require longer chains for accurate estimate."
671+
% draws
672+
)
670673
_log.warning(msg)
671674

672675
auto_nuts_init = True

0 commit comments

Comments
 (0)