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

I’d recommend to compare the variable shapes to see if they are the same between the v3 and v4 models.

If you think there’s a bug with dims-informed shape of a variable, you could pass the explicit shape just to make sure that the RV is not getting resized because of dims.