Thanks for making this.
Which version of pymc was this using? It doesn’t run in 5.8.2, a problem with the concatenate.