token[-1] -> token.rstrip('\n')

2019-07-23 20:41:26 +08:00
parent bef0c629ca
commit b1019d2a8e
1 changed files with 1 additions and 1 deletions
--- a/pytorch_transformers/tokenization_bert.py
+++ b/pytorch_transformers/tokenization_bert.py
@@ -69,7 +69,7 @@ def load_vocab(vocab_file):
    with open(vocab_file, "r", encoding="utf-8") as reader:
        tokens = reader.readlines()
    for index, token in enumerate(tokens):
-        token = token[:-1]
+        token = token.rstrip('\n')
        vocab[token] = index
    return vocab