We dont have a built-in function for that: a workaround is to wrap the comp_dist loglike into a function, something like:
complogp = like.distribution._comp_logp(theano.shared(data))
f_complogp = model.model.fastfn(complogp)
y_ = []
for point in trace:
# get prediction
y_.append(np.argmax(f_complogp(point), axis=1))
Hope this helps!