primeqa.tableqa.utils.data_collator.TapasCollator#

class primeqa.tableqa.utils.data_collator.TapasCollator#

Bases: object

Methods

__call__(batch: List) Dict[str, torch.Tensor]#

Take a list of samples from a Dataset and collate them into a batch. :returns: A dictionary of tensors