Wrong number of args

[aus-covid]

In Realtime Rt mcmc.ipynb there is line which reads
models[state] = create_and_run_model(grp.droplevel(0))

However, the function create_and_run_model expects two arguments
def create_and_run_model(name, state)

Hence, I think the line should read
models[state] = create_and_run_model(state, grp.droplevel(0))