extend works in-place

This commit is contained in:
Rémi Louf committed 2019-10-28 10:49:49 +01:00
1 parent cb26b035c6
commit a67413ccc8
1 file changed
+2 -1
+2 -1
View File
@@ -165,7 +165,8 @@ def _fit_to_block_size(sequence, block_size):
if len(sequence) > block_size: if len(sequence) > block_size:
return sequence[:block_size] return sequence[:block_size]
else: else:
return sequence.extend([0] * (block_size - len(sequence))) sequence.extend([0] * (block_size - len(sequence)))
return sequence
def mask_padding_tokens(sequence): def mask_padding_tokens(sequence):