Commit a67413cc authored by Rémi Louf's avatar Rémi Louf
Browse files

extend works in-place

parent cb26b035
......@@ -165,7 +165,8 @@ def _fit_to_block_size(sequence, block_size):
if len(sequence) > block_size:
return sequence[:block_size]
else:
return sequence.extend([0] * (block_size - len(sequence)))
sequence.extend([0] * (block_size - len(sequence)))
return sequence
def mask_padding_tokens(sequence):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment