How to get original observed data's cluser id or probability after pm.sample by the returned trace object?

Example:Dirichlet process mixtures for density estimation

How to get original observed data’s cluser id after pm.sample by the result trace object?
Thanks anyway.