fixed: vocab.txt missing from package

R1j1t · May 14, 2020 · c1724b0 · c1724b0
1 parent 6c196f7
commit c1724b0
Show file tree

Hide file tree

Showing 6 changed files with 4 additions and 2 deletions.
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -0,0 +1 @@
+graft contextualSpellCheck/data*
diff --git a/contextualSpellCheck/contextualSpellCheck.py b/contextualSpellCheck/contextualSpellCheck.py
@@ -15,7 +15,7 @@ class ContextualSpellCheck(object):
 
     name = "contextual spellchecker"
 
-    def __init__(self, vocab_path="../data/vocab.txt", debug=False):
+    def __init__(self, vocab_path="./data/vocab.txt", debug=False):
         # self.nlp = spacy.load(
         #     "en_core_web_sm", disable=["tagger", "parser"]
         # )  # using default tokeniser with NER

diff --git a/data/vocab.txt → contextualSpellCheck/data/vocab.txt b/data/vocab.txt → contextualSpellCheck/data/vocab.txt
diff --git a/data/bigram_norwig.txt → extra data/bigram_norwig.txt b/data/bigram_norwig.txt → extra data/bigram_norwig.txt
diff --git a/data/spell-errors-testset.txt → extra data/spell-errors-testset.txt b/data/spell-errors-testset.txt → extra data/spell-errors-testset.txt
diff --git a/setup.py b/setup.py
@@ -5,14 +5,15 @@
 
 setuptools.setup(
     name="contextualSpellCheck",
-    version="0.0.1",
+    version="0.0.3",
     author="R1j1t",
     author_email="[email protected]",
     description="Contextual spell correction using BERT (bidirectional representations)",
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/R1j1t/contextualSpellCheck",
     packages=setuptools.find_packages(),
+    include_package_data=True,
     classifiers=[
         "Programming Language :: Python :: 3",
         "Programming Language :: Python :: 3.6",