relative path fix

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

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/contextualSpellCheck/contextualSpellCheck.py b/contextualSpellCheck/contextualSpellCheck.py
@@ -2,6 +2,7 @@
 import torch
 import editdistance
 import datetime
+import os
 
 from spacy.tokens import Doc, Token, Span
 from spacy.vocab import Vocab
@@ -15,7 +16,10 @@ class ContextualSpellCheck(object):
 
     name = "contextual spellchecker"
 
-    def __init__(self, vocab_path="./data/vocab.txt", debug=False):
+    def __init__(self, vocab_path="", debug=False):
+        if vocab_path == "":
+            currentPath = os.path.dirname(__file__)
+            vocab_path = os.path.join(currentPath, 'data/vocab.txt')
         # self.nlp = spacy.load(
         #     "en_core_web_sm", disable=["tagger", "parser"]
         # )  # using default tokeniser with NER

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="contextualSpellCheck",
-    version="0.0.3",
+    version="0.0.4",
     author="R1j1t",
     author_email="r1j1t@protonmail.com",
     description="Contextual spell correction using BERT (bidirectional representations)",