Find clusters in multivariate Iris dataset Dirichlet Process Mixtures in Pymc3

Question

I\'m trying to extend the cluster finding for the Iris dataset finding as described https://www.ritchievink.com/blog/2018/06/05/clustering-data-with-dirichlet-mixtures-in-ed