Update arviz/stats/diagnostics.py

Co-authored-by: Seth Axen <seth@sethaxen.com>
arviz-devs · Nov 12, 2023 · ff971b8 · ff971b8
1 parent ed68f86
commit ff971b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/arviz/stats/diagnostics.py b/arviz/stats/diagnostics.py
@@ -732,7 +732,7 @@ def _ess_sd(ary, relative=False):
     ary = np.asarray(ary)
     if _not_valid(ary, shape_kwargs=dict(min_draws=4, min_chains=1)):
         return np.nan
-    ary = abs(ary - ary.mean())
+    ary = (ary - ary.mean())**2
     return _ess(_split_chains(ary), relative=relative)