fix small bug/typo

364920e2 · VictorSanh · 23c23f53 · 364920e2
Commit 364920e2 authored Sep 10, 2019 by VictorSanh
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/distillation/dataset.py examples/distillation/dataset.py +1 -1

No files found.
--- a/examples/distillation/dataset.py
+++ b/examples/distillation/dataset.py
@@ -77,7 +77,7 @@ class Dataset:
                    if sub_s[0] != cls_id:
                        sub_s = np.insert(sub_s, 0, cls_id)
                    if sub_s[-1] != sep_id:
-                        sub_s = np.insert(sub_s, len(sub_s), cls_id)
+                        sub_s = np.insert(sub_s, len(sub_s), sep_id)
                    assert len(sub_s) <= max_len
                    sub_seqs.append(sub_s)