I tried to generate some samples, get this error at "self.map = sorted([x.getCol().getIndex() for x in self.model.getVars(transformed=True)])".