Arviz plot_energy doesn't like PyMC custom step method?

The arviz energy function seems to not handle traces from multiple NUTS. I am not sure the statistics it is computing make sense with multiple NUTS samplers, perhaps someone like @OriolAbril or @colcarroll will be able to chime in.

1 Like