Time varying survival model (poisson regression) not porting to pymc v4

Questions v5
5

I just double checked, I’m not seeing any missing values, mislabeled groups, or indices without a corresponding coord. The code works fine when running in pymc3 as well