samples = random.sample(self.data[class_], num_shots+1) I'm not sure if I understand it right, can you explain it? thank you very much! best wishes!