extend works in-place

2019-10-17 18:08:09 +02:00
parent cb26b035c6
commit a67413ccc8
1 changed files with 2 additions and 1 deletions
--- a/examples/run_seq2seq_finetuning.py
+++ b/examples/run_seq2seq_finetuning.py
@@ -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):