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