use local import

RasaHQ · Apr 16, 2020 · a013d89 · a013d89
1 parent 4e1b6a6
commit a013d89
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rasa/nlu/test.py b/rasa/nlu/test.py
@@ -37,7 +37,6 @@
 from rasa.nlu.model import Interpreter, Trainer, TrainingData
 from rasa.nlu.components import Component
 from rasa.nlu.tokenizers.tokenizer import Token
-from rasa.nlu.classifiers.diet_classifier import DIETClassifier
 from rasa.utils.tensorflow.constants import ENTITY_RECOGNITION
 
 logger = logging.getLogger(__name__)
@@ -1025,6 +1024,7 @@ def get_entity_extractors(interpreter: Interpreter) -> Set[Text]:
     Processors are removed since they do not detect the boundaries themselves.
     """
     from rasa.nlu.extractors.extractor import EntityExtractor
+    from rasa.nlu.classifiers.diet_classifier import DIETClassifier
 
     extractors = set()
     for c in interpreter.pipeline: