Switch to original fasttext package

The official "fasttext"package has been well maintained. The "fasttextmirror" package might cause installation issues on Windows.
anhaidgroup · Apr 4, 2020 · 5fc2cd4 · 5fc2cd4
1 parent 2efdf87
commit 5fc2cd4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/deepmatcher/data/field.py b/deepmatcher/data/field.py
@@ -6,7 +6,7 @@
 import nltk
 import six
 
-import fastText
+import fasttext
 import torch
 from torchtext import data, vocab
 from torchtext.utils import download_from_url
@@ -77,7 +77,7 @@ def cache(self, name, cache, url=None):
         if not os.path.isfile(path):
             raise RuntimeError('no vectors found at {}'.format(path))
 
-        self.model = fastText.load_model(path)
+        self.model = fasttext.load_model(path)
         self.dim = len(self['a'])
 
 

diff --git a/setup.py b/setup.py
@@ -38,5 +38,5 @@ def find_version(*file_paths):
     python_requires='>=3.5',
     install_requires=[
         'torch>=1.0', 'tqdm', 'pyprind', 'six', 'Cython', 'torchtext', 'nltk>=3.2.5',
-        'fasttextmirror', 'pandas', 'dill', 'scikit-learn'
+        'fasttext', 'pandas', 'dill', 'scikit-learn'
     ])