diff --git a/allennlp/data/data_loaders/data_loader.py b/allennlp/data/data_loaders/data_loader.py index 912a8b6b666..3724e3f849b 100644 --- a/allennlp/data/data_loaders/data_loader.py +++ b/allennlp/data/data_loaders/data_loader.py @@ -20,7 +20,7 @@ def allennlp_collate(instances: List[Instance]) -> TensorDict: batch. """ batch = Batch(instances) - return batch.as_tensor_dict(batch.get_padding_lengths()) + return batch.as_tensor_dict() class DataLoader(Registrable):