Arviz plot_energy doesn't like PyMC custom step method?

11

Thanks @OriolAbril, I will probably lean more towards the former option of diagnosing each nuts sampler independently but I may try out both codes provided and have a bit of a read/play myself to see about how the energy information of multiple samplers is added together like you point out.

1 Like