diff --git a/src/miade/core.py b/src/miade/core.py index 76d2d85..91d109f 100644 --- a/src/miade/core.py +++ b/src/miade/core.py @@ -113,7 +113,9 @@ def _load_model_factory(self, custom_annotators: Optional[List[Annotator]] = Non try: cat = MiADE_CAT.load_model_pack(str(model_pack_filepath), meta_cat_config_dict=meta_cat_config_dict) # temp fix reload to load stop words - cat.pipe._nlp = spacy.load(cat.config.general.spacy_model, disable=cat.config.general.spacy_disabled_components) + cat.pipe._nlp = spacy.load( + cat.config.general.spacy_model, disable=cat.config.general.spacy_disabled_components + ) cat._create_pipeline(config=cat.config) cat_id = cat.config.version["id"] loaded_models[cat_id] = cat diff --git a/src/miade/utils/miade_cat.py b/src/miade/utils/miade_cat.py index 9f67885..aef9190 100644 --- a/src/miade/utils/miade_cat.py +++ b/src/miade/utils/miade_cat.py @@ -301,4 +301,4 @@ def train_supervised( # Set the filters again self.config.linking["filters"] = _filters - return fp, fn, tp, p, r, f1, cui_counts, examples \ No newline at end of file + return fp, fn, tp, p, r, f1, cui_counts, examples