from arviz_base import load_arviz_data

import arviz_plots as azp

azp.style.use("arviz-variat")

centered = load_arviz_data("centered_eight")
non_centered = load_arviz_data("non_centered_eight")

pc = azp.plot_ridge(
    {
        "centered": centered,
        "non-centered": non_centered
    },
    coords={"school":["Deerfield", "St. Paul's", "Hotchkiss"]},
    backend="none" # change to preferred backend
)
pc.add_legend("model")
pc.show()