Update Translator.py

linguisticsai · Mar 27, 2018 · c416bc4 · c416bc4
1 parent 969b6d0
commit c416bc4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/onmt/translate/Translator.py b/onmt/translate/Translator.py
@@ -80,7 +80,8 @@ def translate_batch(self, batch, data):
 
         # Define a list of tokens to exclude from ngram-blocking
         # exclusion_list = ["<t>", "</t>", "."]
-        exclusion_tokens = set([vocab.stoi[t] for t in self.ignore_when_blocking])
+        exclusion_tokens = set([vocab.stoi[t] 
+                                for t in self.ignore_when_blocking])
 
         beam = [onmt.translate.Beam(beam_size, n_best=self.n_best,
                                     cuda=self.cuda,