Update namedtuple

PetrochukM · Oct 21, 2019 · f1b8144 · f1b8144
1 parent 0924ec7
commit f1b8144
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchnlp/encoders/text/text_encoder.py b/torchnlp/encoders/text/text_encoder.py
@@ -25,7 +25,7 @@ def pad_tensor(tensor, length, padding_index=DEFAULT_PADDING_INDEX):
     return torch.cat((tensor, padding), dim=0)
 
 
-BatchedSequences = namedtuple('BatchedSequences', ['sequences', 'lengths'])
+BatchedSequences = namedtuple('BatchedSequences', ['tensor', 'lengths'])
 
 
 def stack_and_pad_tensors(batch, padding_index=DEFAULT_PADDING_INDEX, dim=0):