Skip to content

GitLab

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

extend works in-place

parent cb26b035
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
examples/run_seq2seq_finetuning.py
View file @ a67413cc
...@@ -165,7 +165,8 @@ def _fit_to_block_size(sequence, block_size): ...@@ -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):
......
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