Are you sure you have the last version of pymc-experimental? I tried with a silly example and it works, in my example y_train has shape (100,2) and Bart shape (2, 100). I will try with a more realistic example, it will be a good exercise to see how BART works with categorical distributions.