Skip to content

Commit

Permalink
fix issue with numba version
Browse files Browse the repository at this point in the history
  • Loading branch information
OriolAbril committed Mar 23, 2022
1 parent 67891bc commit ba2809f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion arviz/stats/diagnostics.py
Original file line number Diff line number Diff line change
Expand Up @@ -829,7 +829,7 @@ def _mcse_mean(ary):
return np.nan
ess = _ess_mean(ary)
if _numba_flag:
sd = _sqrt(svar(np.ravel(ary), ddof=1), np.zeros(1))
sd = _sqrt(svar(np.ravel(ary), ddof=1), 0)
else:
sd = np.std(ary, ddof=1)
mcse_mean_value = sd / np.sqrt(ess)
Expand Down

0 comments on commit ba2809f

Please sign in to comment.