Skip to content

Commit

Permalink
start ess related improvements
Browse files Browse the repository at this point in the history
  • Loading branch information
OriolAbril committed Dec 24, 2024
1 parent b6f254f commit c8606a2
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions src/arviz_stats/base/dataarray.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,6 +57,7 @@ def hdi(self, da, prob=None, dims=None, method="nearest", **kwargs):
def ess(self, da, dims=None, method="bulk", relative=False, prob=None):
"""Compute ess on DataArray input."""
dims, chain_axis, draw_axis = validate_dims_chain_draw_axis(dims)
prob = validate_ci_prob(prob)
return apply_ufunc(
self.array_class.ess,
da,
Expand Down

0 comments on commit c8606a2

Please sign in to comment.