Sorry if I wasn’t clear but it would be helpful to debug if you could provide a small example that I can run which produces the error. It seems that __getitem__ returns something odd and I’m guessing it would be the label but it’s hard to debug without any sample code.