def load():
import medspacy
nlp = medspacy.load(enable=["sentencizer", "tokenizer"])
# Add components
from medspacy.target_matcher import TargetMatcher, TargetRule
target_matcher = TargetMatcher(nlp)
target_filepath = path.join(RESOURCES_DIR, "target_rules.json")
target_rules = TargetRule.from_json(target_filepath)
target_matcher.add(target_rules)
nlp.add_pipe(target_matcher)
from medspacy.context import ConTextComponent, ConTextRule
context_filepath = path.join(RESOURCES_DIR, "context_rules.json")
context = ConTextComponent(nlp, rules=None, add_attrs=CONTEXT_ATTRS)
context_rules = ConTextRule.from_json(context_filepath)
context.add(context_rules)
nlp.add_pipe(context)
from medspacy.section_detection import Sectionizer
# TODO: Add radiology section rules
sectionizer = Sectionizer(nlp)
nlp.add_pipe(sectionizer)
clf = DocumentClassifier(nlp)
nlp.add_pipe(clf)
return nlp
def test_from_json(self, from_json_file):
assert TargetRule.from_json(from_json_file)