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

There was a small typo in my example (edited). Here it is running on Google Colab: Google Colab

If you are using an old version of PyMC do update it if you can.