def __getitem__(self, idx):
    item = {k: torch.tensor(v[idx]) for (k, v) in self.encodings.items()}
    item['labels'] = torch.tensor([self.labels[idx]])
    return item
def __getitem__(self, idx):
    item = {k: torch.tensor(v[idx]) for (k, v) in self.encodings.items()}
    item['labels'] = torch.tensor([self.labels[idx]])
    return item