Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
ruff format
Browse files Browse the repository at this point in the history
jenniferajiang committed Dec 18, 2023
1 parent 92b3c5f commit cd4684f
Showing 2 changed files with 4 additions and 2 deletions.
4 changes: 3 additions & 1 deletion src/miade/core.py
Original file line number Diff line number Diff line change
@@ -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
2 changes: 1 addition & 1 deletion src/miade/utils/miade_cat.py
Original file line number Diff line number Diff line change
@@ -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
return fp, fn, tp, p, r, f1, cui_counts, examples

0 comments on commit cd4684f

Please sign in to comment.