got a stack trace with this. I fixed the strata_samples to use the class labels, but that doesn't seem to be the issue?
https://0xdata.atlassian.net/browse/HEX-593