fix typo in previous commit

rsennrich · Jun 28, 2018 · 48ba99e · 48ba99e
1 parent 61ad855
commit 48ba99e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/subword_nmt/learn_bpe.py b/subword_nmt/learn_bpe.py
@@ -221,7 +221,7 @@ def learn_bpe(infile, outfile, num_symbols, min_frequency=2, verbose=False, is_d
         for word in vocab:
             for char in word[:-1]:
                 uniq_char_internal.add(char)
-            uniq_char_final.add(char[-1])
+            uniq_char_final.add(word[-1])
         sys.stderr.write('Number of word-internal characters: {0}\n'.format(len(uniq_char_internal)))
         sys.stderr.write('Number of word-final characters: {0}\n'.format(len(uniq_char_final)))
         sys.stderr.write('Reducing number of merge operations by {0}\n'.format(len(uniq_char_internal) + len(uniq_char_final)))