BUG clip effective sample size at maximum number of samples

beckermr · web-flow · commit 8d8413889767 · 2016-09-29T17:10:33.000-05:00
diff --git a/pymc3/diagnostics.py b/pymc3/diagnostics.py
@@ -259,7 +259,7 @@ def calc_n_eff(x):
 
             t += 1
 
-        return int(m * n / (1. + 2 * rho[1:t].sum()))
+        return min(m * n, int(m * n / (1. + 2 * rho[1:t].sum())))
 
     n_eff = {}
     for var in mtrace.varnames:
