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

Thanks @ricardoV94. Not sure I understand your point, I do have separate index for testing versus training

idx = temp_df.race_idx

and

idx2 = temp_df2.race_idx

Isn’t this the right way to do it?