Moved extra init code to a .py file

aajanki · Aug 28, 2020 · d9e4824 · d9e4824
1 parent a261fa5
commit d9e4824
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 11 deletions.
diff --git a/python_packaging/init_extra.py b/python_packaging/init_extra.py
@@ -0,0 +1,8 @@
+
+
+
+from spacy.util import set_lang_class
+from .fi import FinnishEx, FinnishExDefaults
+from .lemmatizer import FinnishLemmatizer
+
+set_lang_class('fi', FinnishEx)
diff --git a/tools/package_model.sh b/tools/package_model.sh
@@ -19,17 +19,7 @@ mkdir -p models/python-package/"$PACKAGE_DIR/$MODEL_NAME"
 cp -r fi/[^_]*.py fi/lookups/ models/python-package/"$PACKAGE_DIR/$MODEL_NAME"/
 
 echo "Adding import to __init__.py"
-cat - >> models/python-package/"$PACKAGE_DIR/$MODEL_NAME"/__init__.py <<EOF
-
-
-
-from spacy.util import set_lang_class
-from .fi import FinnishEx, FinnishExDefaults
-from .lemmatizer import FinnishLemmatizer
-
-set_lang_class('fi', FinnishEx)
-EOF
-
+cat python_packaging/init_extra.py >> models/python-package/"$PACKAGE_DIR/$MODEL_NAME"/__init__.py
 
 echo "Updating requirements in meta.json"
 jq '.requirements = ["voikko>=0.5"]' \