explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-26 09:56:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

Avoid forcing a name on empty vectors, and remove print statement

Browse Source
This commit is contained in:
Matthew Honnibal 2018-03-28 21:08:58 +02:00
parent cf5fcf0546
commit a7c5ae2beb
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/language.py
View File

@ -695,6 +695,8 @@ def _fix_pretrained_vectors_name(nlp):
# data # data
if 'vectors' in nlp.meta and nlp.meta['vectors'].get('name'): if 'vectors' in nlp.meta and nlp.meta['vectors'].get('name'):
nlp.vocab.vectors.name = nlp.meta['vectors']['name'] nlp.vocab.vectors.name = nlp.meta['vectors']['name']
elif not nlp.vocab.vectors.size:
nlp.vocab.vectors.name = None
elif 'name' in nlp.meta and 'lang' in nlp.meta: elif 'name' in nlp.meta and 'lang' in nlp.meta:
vectors_name = '%s_%s.vectors' % (nlp.meta['lang'], nlp.meta['name']) vectors_name = '%s_%s.vectors' % (nlp.meta['lang'], nlp.meta['name'])
nlp.vocab.vectors.name = vectors_name nlp.vocab.vectors.name = vectors_name
@ -706,7 +708,6 @@ def _fix_pretrained_vectors_name(nlp):
if proc.cfg.get('pretrained_dims'): if proc.cfg.get('pretrained_dims'):
assert nlp.vocab.vectors.name assert nlp.vocab.vectors.name
proc.cfg['pretrained_vectors'] = nlp.vocab.vectors.name proc.cfg['pretrained_vectors'] = nlp.vocab.vectors.name
print(proc.cfg)
class DisabledPipes(list): class DisabledPipes(list):