diff --git a/setup.cfg b/setup.cfg --- a/setup.cfg +++ b/setup.cfg @@ -18,3 +18,6 @@ packages = package_dir = =src python_requires = >=3.6 + +[options.package_data] +languedoc = models.json.gz diff --git a/src/languedoc/predict.py b/src/languedoc/predict.py --- a/src/languedoc/predict.py +++ b/src/languedoc/predict.py @@ -5,7 +5,7 @@ import json import gzip TOP_NGRAM_COUNT = 3000 -MODEL_PATH = os.path.join(os.path.dirname(__file__), "../../models.json.gz") +MODEL_PATH = os.path.join(os.path.dirname(__file__), "models.json.gz") def preprocess(text): diff --git a/src/languedoc/train.py b/src/languedoc/train.py --- a/src/languedoc/train.py +++ b/src/languedoc/train.py @@ -74,7 +74,7 @@ def cross_validate(sample_sets): DATA_DIR = os.path.join(os.path.dirname(__file__), "../../data") LANG_DIRS = sorted([x.path for x in os.scandir(DATA_DIR)]) -MODEL_PATH = os.path.join(os.path.dirname(__file__), "../../models.json.gz") +MODEL_PATH = os.path.join(os.path.dirname(__file__), "models.json.gz") if __name__ == "__main__": samples = []