LDA sampling code with PyMC