How to properly do out-of-sample prediction for hierarchical model

Slowly losing hope; the additional model for prediction is still showing

TypeError: ("The type's shape ((7447,)) is not compatible with the data's ((5949,))", 'Container name "b0"')

where 7447 is the overall number of groups in the full data set; 5949 is the number of groups in the training model.

I thought this thread
https://discourse.pymc.io/t/out-of-sample-prediction/11794

is relevant to my issue but not sure how to apply the solution.