from arviz_base import load_arviz_data

import arviz_plots as azp

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

dt = load_arviz_data("rugby")
pc = azp.plot_ppc_rootogram(
    dt,
    aes={"color": ["__variable__"]}, # map variable to color
    aes_by_visuals={"title": ["color"]}, # change title's color per variable
    backend="none",
)
pc.show()