Something wrong in the code from `PyMC 4.0 with labeled coords and dims`

@OriolAbril