🐛 Fix import cleanup regressions

uclh-criu · Nov 21, 2023 · 0ca5604 · 0ca5604
1 parent edbfdbf
commit 0ca5604
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/miade/core.py b/src/miade/core.py
@@ -3,12 +3,13 @@
 import yaml
 import logging
 
+from negspacy.negation import Negex  # noqa: F401
 from pathlib import Path
 from typing import List, Optional, Dict
 
 from .concept import Concept, Category
 from .note import Note
-from .annotators import Annotator
+from .annotators import Annotator, ProblemsAnnotator, MedsAllergiesAnnotator  # noqa: F401
 from .dosageextractor import DosageExtractor
 from .utils.miade_cat import MiADE_CAT
 from .utils.modelfactory import ModelFactory

diff --git a/src/miade/dosageextractor.py b/src/miade/dosageextractor.py
@@ -5,7 +5,9 @@
 from typing import Optional
 
 from .dosage import Dosage
-
+from .drugdoseade.preprocessor import Preprocessor  # noqa: F401
+from .drugdoseade.pattern_matcher import PatternMatcher  # noqa: F401
+from .drugdoseade.entities_refiner import EntitiesRefiner  # noqa: F401
 
 log = logging.getLogger(__name__)