PyMC3 posterior prediction example; how to incorporate a matrix of betas?

[OriolAbril]

Should be something like

treatment_data = pm.Data('treatment_data', treatment[:, None], dims=('obs_id', "treatment"))