Reduce number of minimum draw in warning message

ricardoV94 · Feb 20, 2024 · 97a7a00 · 97a7a00
1 parent 7b38a48
commit 97a7a00
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pymc/sampling/mcmc.py b/pymc/sampling/mcmc.py
@@ -665,8 +665,11 @@ def sample(
     if draws == 0:
         msg = "Tuning was enabled throughout the whole trace."
         _log.warning(msg)
-    elif draws < 500:
-        msg = "Only %s samples in chain." % draws
+    elif draws < 100:
+        msg = (
+            "Only %s samples per chain. Reliable r-hat and ESS diagnostics require longer chains for accurate estimate."
+            % draws
+        )
         _log.warning(msg)
 
     auto_nuts_init = True