Time-series regression - shape error / Input dimension mis-match

Thank you.

PYMC and aesara run fine now, but I am still trying to get the model to work - I think the issue is related to the hierarchical nature of my data. I’ll start a separate post on this.