Skip to content

Commit 0e6658c

Browse files
new: Add split_parse model config to setup.py
1 parent 6db7d8e commit 0e6658c

File tree

3 files changed

+9
-5
lines changed

3 files changed

+9
-5
lines changed

deep_reference_parser/__version__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,4 +7,4 @@
77
__license__ = "MIT"
88
__splitter_model_version__ = "2020.3.6_splitting"
99
__parser_model_version__ = "2020.3.8_parsing"
10-
__multitask_model_version__ = "2020.3.19_multitask"
10+
__splitparser_model_version__ = "2020.3.19_multitask"

deep_reference_parser/common.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,12 @@
55
from logging import getLogger
66
from urllib import parse, request
77

8+
from .__version__ import (
9+
__parser_model_version__,
10+
__splitparser_model_version__,
11+
__splitter_model_version__,
12+
)
813
from .logger import logger
9-
from .__version__ import __splitter_model_version__, __parser_model_version__, __multitask_model_version__
1014

1115

1216
def get_path(path):
@@ -15,7 +19,7 @@ def get_path(path):
1519

1620
SPLITTER_CFG = get_path(f"configs/{__splitter_model_version__}.ini")
1721
PARSER_CFG = get_path(f"configs/{__parser_model_version__}.ini")
18-
MULTITASK_CFG = get_path(f"configs/{__multitask_model_version__}.ini")
22+
MULTITASK_CFG = get_path(f"configs/{__splitparser_model_version__}.ini")
1923

2024

2125
def download_model_artefact(artefact, s3_slug):
@@ -48,8 +52,7 @@ def download_model_artefacts(model_dir, s3_slug, artefacts=None):
4852
if not artefacts:
4953

5054
artefacts = [
51-
"indices.pickle"
52-
"maxes.pickle",
55+
"indices.pickle" "maxes.pickle",
5356
"weights.h5",
5457
]
5558

setup.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@
3535
"deep_reference_parser": [
3636
f"configs/{about['__splitter_model_version__']}.ini",
3737
f"configs/{about['__parser_model_version__']}.ini",
38+
f"configs/{about['__splitparser_model_version__']}.ini",
3839
]
3940
},
4041
classifiers=[

0 commit comments

Comments
 (0)