Skip to content

Commit

Permalink
Oops, that 'splains the charlm errors for the nocharlm and biomed dep…
Browse files Browse the repository at this point in the history
…parse models
  • Loading branch information
AngledLuffa committed Aug 23, 2023
1 parent 5804ffc commit 09aba46
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions stanza/resources/prepare_resources.py
Original file line number Diff line number Diff line change
Expand Up @@ -244,10 +244,7 @@ def get_depparse_dependencies(lang, package):
else:
dependencies = [{'model': 'pretrain', 'package': package}]

if lang in depparse_charlms and package in depparse_charlms[lang]:
charlm_package = depparse_charlms[lang][package]
else:
charlm_package = default_charlms.get(lang, None)
charlm_package = get_depparse_charlm_package(lang, package)

if charlm_package is not None:
dependencies.append({'model': 'forward_charlm', 'package': charlm_package})
Expand All @@ -267,7 +264,10 @@ def get_ner_dependencies(lang, package):
if pretrain_package is not None:
dependencies = [{'model': 'pretrain', 'package': pretrain_package}]

charlm_package = get_depparse_charlm_package(lang, package)
if lang not in ner_charlms or package not in ner_charlms[lang]:
charlm_package = default_charlms.get(lang, None)
else:
charlm_package = ner_charlms[lang][package]

if charlm_package is not None:
dependencies = dependencies + [{'model': 'forward_charlm', 'package': charlm_package},
Expand Down

0 comments on commit 09aba46

Please sign in to comment.