From 08382f21e30ffd38f5905764c01e2ae66787e3d0 Mon Sep 17 00:00:00 2001 From: ines Date: Mon, 29 May 2017 20:44:11 +0200 Subject: [PATCH] Pass model meta to nlp object in load_model --- spacy/util.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/spacy/util.py b/spacy/util.py index 329c6ce88..00bc4e9fc 100644 --- a/spacy/util.py +++ b/spacy/util.py @@ -108,12 +108,12 @@ def load_model(name, **overrides): model_path = Path(name) meta = get_package_meta(model_path) 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) elif hasattr(name, 'exists'): # Path or Path-like to model data meta = get_package_meta(name) 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) 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(): raise ValueError("Can't find model directory: %s" % path2str(data_path)) 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) @@ -461,7 +461,7 @@ def model_from_bytes(model, bytes_data): i += 1 if hasattr(layer, '_layers'): queue.extend(layer._layers) - + def print_table(data, title=None): """Print data in table format.