Discard one chain from mcmc object

numpyro
1

Is there a way to exclude one chain from the mcmc object since this once chain seems to get stuck (see plot below)?

image

2

I think you can do

samples = mcmc.get_samples(group_by_chain=True)
filtered_samples = {k: v[1:] for k, v in samples.items()}