Pass model meta to nlp object in load_model

This commit is contained in:
ines 2017-05-29 20:44:11 +02:00
parent 6145fe6a93
commit 08382f21e3
1 changed files with 4 additions and 4 deletions

View File

@ -108,12 +108,12 @@ def load_model(name, **overrides):
model_path = Path(name) model_path = Path(name)
meta = get_package_meta(model_path) meta = get_package_meta(model_path)
cls = get_lang_class(meta['lang']) cls = get_lang_class(meta['lang'])
nlp = cls(pipeline=meta.get('pipeline', True)) nlp = cls(pipeline=meta.get('pipeline', True), meta=meta)
return nlp.from_disk(model_path, **overrides) return nlp.from_disk(model_path, **overrides)
elif hasattr(name, 'exists'): # Path or Path-like to model data elif hasattr(name, 'exists'): # Path or Path-like to model data
meta = get_package_meta(name) meta = get_package_meta(name)
cls = get_lang_class(meta['lang']) cls = get_lang_class(meta['lang'])
nlp = cls(pipeline=meta.get('pipeline', True)) nlp = cls(pipeline=meta.get('pipeline', True), meta=meta)
return nlp.from_disk(name, **overrides) return nlp.from_disk(name, **overrides)
raise IOError("Can't find model '%s'" % name) raise IOError("Can't find model '%s'" % name)
@ -133,7 +133,7 @@ def load_model_from_init_py(init_file, **overrides):
if not model_path.exists(): if not model_path.exists():
raise ValueError("Can't find model directory: %s" % path2str(data_path)) raise ValueError("Can't find model directory: %s" % path2str(data_path))
cls = get_lang_class(meta['lang']) cls = get_lang_class(meta['lang'])
nlp = cls(pipeline=meta.get('pipeline', True)) nlp = cls(pipeline=meta.get('pipeline', True), meta=meta)
return nlp.from_disk(data_path, **overrides) return nlp.from_disk(data_path, **overrides)