This notebooks is unfortunately quite out of date. You can make this plot in one line using arviz.plot_pair:
import arviz as az
az.plot_pair(short_trace,
var_names=['theta', 'mu'],
coords={'theta_dim_0':[0]},
divergences=True)
The readability can be improved by adding coords to the model, as shown in this notebook.