diff options
| author | John Bauer <horatio@gmail.com> | 2022-09-13 23:22:07 +0300 |
|---|---|---|
| committer | John Bauer <horatio@gmail.com> | 2022-09-13 23:22:07 +0300 |
| commit | 06f95b4222918de0baecca34cd1769068124a498 (patch) | |
| tree | c664d7bc79499855777bac322b41d56662ff9b47 | |
| parent | a4dfccbc82c3e6fe629d79fa63ccdea176c472eb (diff) | |
Temporarily don't include charlm in the POS models for PL - those haven't retrained yet
| -rw-r--r-- | stanza/resources/prepare_resources.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/stanza/resources/prepare_resources.py b/stanza/resources/prepare_resources.py index 94dc01cf..055038f9 100644 --- a/stanza/resources/prepare_resources.py +++ b/stanza/resources/prepare_resources.py @@ -190,6 +190,7 @@ pos_charlms = { "genia": None, "mimic": "mimic", }, + "pl": None, # temporary until these are rebuilt with charlm "tr": { # no idea why, but this particular one goes down in dev score "boun": None, }, @@ -372,8 +373,10 @@ def get_pos_dependencies(lang, package): else: dependencies = [{'model': 'pretrain', 'package': package}] - if lang in pos_charlms and package in pos_charlms[lang]: + if lang in pos_charlms and isinstance(pos_charlms[lang], dict) and package in pos_charlms[lang]: charlm_package = pos_charlms[lang][package] + elif lang in pos_charlms: + charlm_package = pos_charlms[lang] else: charlm_package = default_charlms.get(lang, None) |