That is how I would do it as well but I think op wants some arviz defined parameter to turn it off, which does not seem to exist
I am looking into how to do this by a combination of >using ax.set_yticks() and ax.axvlilne() but there should >be something more straightforward.