The issue now is that from_pymc3_predictions takes the coordinates from the model, which have not been updated and therefore still have the original obs_id length. You have to provide new coordinates, something like as.from_pymc3_predictions(..., coords={"obs_id": [1, 2, 3]})