By using shared variables with theano I managed to solve my issue thanks to this post: Prediction using sample_ppc in Hierarchical model
By using shared variables with theano I managed to solve my issue thanks to this post: Prediction using sample_ppc in Hierarchical model