KeyError Setting Mixture Proportions for Mixture Model

Try setting the dtype like this

If that doesn’t work, it may be related to an issue we’re looking into but haven’t fully identified that slowed our run times.