Multi-level model with nonsense state and county coefficients

try:

az.plot_forest(
     trace, var_names="a_county_offset", figsize=(12, 6), r_hat=True, combined=True, textsize=16
);

a_county is expanded to the length of obs; the shape of a_county_offset is the # of counties