You have to make sure that all the shapes work together. Things that come to mind and would be issues:
- You are creating
pm.Databut not using it later in the model, so updating the data won’t change anything in the model, only what ArviZ stored asconstant_data - You are only updating only treatment and not covariates, those are added, so their shapes need to match for the model to work.