# HG changeset patch
# User Laman
# Date 2022-10-17 19:12:22
# Node ID 252d3b1bca6051198452fee201d4c88202712b5f
# Parent  f96c1c4d0bc9c615ec361402bafd9c589dd50e70

models file included in the package

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 = []