Get subword-nmt from PyPI

lvapeab · lvapeab · commit 32d5c0e171ee · 2020-04-21T09:59:06.000+02:00
diff --git a/keras_wrapper/dataset.py b/keras_wrapper/dataset.py
@@ -2030,9 +2030,13 @@ def build_bpe(self, codes, merges=-1, separator=u'@@', vocabulary=None, glossari
                            nor concatenated with other subwords.
         :return: None
         """
-        from keras_wrapper.extra.external import BPE
+        from subword_nmt.apply_bpe import BPE
         with codecs.open(codes, 'rb', encoding='utf-8') as cods:
-            self.BPE = BPE(cods, merges=merges, separator=separator, vocab=vocabulary, glossaries=glossaries)
+            self.BPE = BPE(cods,
+                           merges=merges,
+                           separator=separator,
+                           vocab=vocabulary,
+                           glossaries=glossaries)
         self.BPE_separator = separator
         self.BPE_built = True
 
diff --git a/keras_wrapper/extra/external.py b/keras_wrapper/extra/external.py
diff --git a/setup.py b/setup.py
@@ -32,6 +32,7 @@
                         'sacremoses',
                         'sacrebleu',
                         'scipy',
+                        'subword-nmt',
                         'future',
                         'cython',
                         'keras_applications',
