def __call__(self, features: List[Dict[(str, Union[(List[int], torch.Tensor)])]]) -> Dict[(str, torch.Tensor)]:
    batch = self.tokenizer.pad(features, padding=self.padding, max_length=self.max_length, pad_to_multiple_of=self.pad_to_multiple_of, return_tensors='pt')
    if ('label' in batch):
        batch['labels'] = batch['label']
        del batch['label']
    if ('label_ids' in batch):
        batch['labels'] = batch['label_ids']
        del batch['label_ids']
    return batch
def __call__(self, features: List[Dict[(str, Union[(List[int], torch.Tensor)])]]) -> Dict[(str, torch.Tensor)]:
    batch = self.tokenizer.pad(features, padding=self.padding, max_length=self.max_length, pad_to_multiple_of=self.pad_to_multiple_of, return_tensors='pt')
    if ('label' in batch):
        batch['labels'] = batch['label']
        del batch['label']
    if ('label_ids' in batch):
        batch['labels'] = batch['label_ids']
        del batch['label_ids']
    return batch