rasa-pro 3.12.0.dev2__py3-none-any.whl → 3.12.0.dev4__py3-none-any.whl

rasa/nlu/classifiers/diet_classifier.py

@@ -4,112 +4,111 @@ import copy
 import logging
 from collections import defaultdict
 from pathlib import Path
-from typing import Any, Dict, List, Optional, Text, Tuple, Union, TypeVar, Type
+from typing import Any, Dict, List, Optional, Text, Tuple, Type, TypeVar, Union
 
 import numpy as np
 import scipy.sparse
 import tensorflow as tf
 
-from rasa.exceptions import ModelNotFound
-from rasa.nlu.featurizers.featurizer import Featurizer
+import rasa.nlu.utils.bilou_utils as bilou_utils
+import rasa.shared.utils.io
 from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.recipes.default_recipe import DefaultV1Recipe
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
-from rasa.nlu.extractors.extractor import EntityExtractorMixin
+from rasa.exceptions import ModelNotFound
+from rasa.nlu.classifiers import LABEL_RANKING_LENGTH
 from rasa.nlu.classifiers.classifier import IntentClassifier
-import rasa.shared.utils.io
-import rasa.nlu.utils.bilou_utils as bilou_utils
+from rasa.nlu.constants import DEFAULT_TRANSFORMER_SIZE, TOKENS_NAMES
+from rasa.nlu.extractors.extractor import EntityExtractorMixin, EntityTagSpec
+from rasa.nlu.featurizers.featurizer import Featurizer
 from rasa.shared.constants import DIAGNOSTIC_DATA
-from rasa.nlu.extractors.extractor import EntityTagSpec
-from rasa.nlu.classifiers import LABEL_RANKING_LENGTH
-from rasa.utils import train_utils
-from rasa.utils.tensorflow import rasa_layers
-from rasa.utils.tensorflow.feature_array import (
-    FeatureArray,
-    serialize_nested_feature_arrays,
-    deserialize_nested_feature_arrays,
-)
-from rasa.utils.tensorflow.models import RasaModel, TransformerRasaModel
-from rasa.utils.tensorflow.model_data import (
-    RasaModelData,
-    FeatureSignature,
-)
-from rasa.nlu.constants import TOKENS_NAMES, DEFAULT_TRANSFORMER_SIZE
+from rasa.shared.exceptions import InvalidConfigException
 from rasa.shared.nlu.constants import (
-    SPLIT_ENTITIES_BY_COMMA_DEFAULT_VALUE,
-    TEXT,
-    INTENT,
-    INTENT_RESPONSE_KEY,
     ENTITIES,
-    ENTITY_ATTRIBUTE_TYPE,
     ENTITY_ATTRIBUTE_GROUP,
     ENTITY_ATTRIBUTE_ROLE,
+    ENTITY_ATTRIBUTE_TYPE,
+    INTENT,
+    INTENT_RESPONSE_KEY,
     NO_ENTITY_TAG,
     SPLIT_ENTITIES_BY_COMMA,
+    SPLIT_ENTITIES_BY_COMMA_DEFAULT_VALUE,
+    TEXT,
 )
-from rasa.shared.exceptions import InvalidConfigException
-from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.shared.nlu.training_data.message import Message
+from rasa.shared.nlu.training_data.training_data import TrainingData
+from rasa.utils import train_utils
+from rasa.utils.tensorflow import rasa_layers
 from rasa.utils.tensorflow.constants import (
-    DROP_SMALL_LAST_BATCH,
-    LABEL,
-    IDS,
-    HIDDEN_LAYERS_SIZES,
-    RENORMALIZE_CONFIDENCES,
-    SHARE_HIDDEN_LAYERS,
-    TRANSFORMER_SIZE,
-    NUM_TRANSFORMER_LAYERS,
-    NUM_HEADS,
+    AUTO,
+    BALANCED,
     BATCH_SIZES,
     BATCH_STRATEGY,
+    BILOU_FLAG,
+    CHECKPOINT_MODEL,
+    CONCAT_DIMENSION,
+    CONNECTION_DENSITY,
+    CONSTRAIN_SIMILARITIES,
+    CROSS_ENTROPY,
+    DENSE_DIMENSION,
+    DENSE_INPUT_DROPOUT,
+    DROP_RATE,
+    DROP_RATE_ATTENTION,
+    DROP_SMALL_LAST_BATCH,
+    EMBEDDING_DIMENSION,
+    ENTITY_RECOGNITION,
     EPOCHS,
-    RANDOM_SEED,
+    EVAL_NUM_EPOCHS,
+    EVAL_NUM_EXAMPLES,
+    FEATURIZERS,
+    HIDDEN_LAYERS_SIZES,
+    IDS,
+    INTENT_CLASSIFICATION,
+    KEY_RELATIVE_ATTENTION,
+    LABEL,
     LEARNING_RATE,
-    RANKING_LENGTH,
     LOSS_TYPE,
-    SIMILARITY_TYPE,
-    NUM_NEG,
-    SPARSE_INPUT_DROPOUT,
-    DENSE_INPUT_DROPOUT,
+    MASK,
     MASKED_LM,
-    ENTITY_RECOGNITION,
-    TENSORBOARD_LOG_DIR,
-    INTENT_CLASSIFICATION,
-    EVAL_NUM_EXAMPLES,
-    EVAL_NUM_EPOCHS,
-    UNIDIRECTIONAL_ENCODER,
-    DROP_RATE,
-    DROP_RATE_ATTENTION,
-    CONNECTION_DENSITY,
-    NEGATIVE_MARGIN_SCALE,
-    REGULARIZATION_CONSTANT,
-    SCALE_LOSS,
-    USE_MAX_NEG_SIM,
     MAX_NEG_SIM,
     MAX_POS_SIM,
-    EMBEDDING_DIMENSION,
-    BILOU_FLAG,
-    KEY_RELATIVE_ATTENTION,
-    VALUE_RELATIVE_ATTENTION,
     MAX_RELATIVE_POSITION,
-    AUTO,
-    BALANCED,
-    CROSS_ENTROPY,
-    TENSORBOARD_LOG_LEVEL,
-    CONCAT_DIMENSION,
-    FEATURIZERS,
-    CHECKPOINT_MODEL,
-    SEQUENCE,
+    MODEL_CONFIDENCE,
+    NEGATIVE_MARGIN_SCALE,
+    NUM_HEADS,
+    NUM_NEG,
+    NUM_TRANSFORMER_LAYERS,
+    RANDOM_SEED,
+    RANKING_LENGTH,
+    REGULARIZATION_CONSTANT,
+    RENORMALIZE_CONFIDENCES,
+    RUN_EAGERLY,
+    SCALE_LOSS,
     SENTENCE,
+    SEQUENCE,
     SEQUENCE_LENGTH,
-    DENSE_DIMENSION,
-    MASK,
-    CONSTRAIN_SIMILARITIES,
-    MODEL_CONFIDENCE,
+    SHARE_HIDDEN_LAYERS,
+    SIMILARITY_TYPE,
     SOFTMAX,
-    RUN_EAGERLY,
+    SPARSE_INPUT_DROPOUT,
+    TENSORBOARD_LOG_DIR,
+    TENSORBOARD_LOG_LEVEL,
+    TRANSFORMER_SIZE,
+    UNIDIRECTIONAL_ENCODER,
+    USE_MAX_NEG_SIM,
+    VALUE_RELATIVE_ATTENTION,
 )
+from rasa.utils.tensorflow.feature_array import (
+    FeatureArray,
+    deserialize_nested_feature_arrays,
+    serialize_nested_feature_arrays,
+)
+from rasa.utils.tensorflow.model_data import (
+    FeatureSignature,
+    RasaModelData,
+)
+from rasa.utils.tensorflow.models import RasaModel, TransformerRasaModel
 
 logger = logging.getLogger(__name__)
 

rasa/nlu/classifiers/fallback_classifier.py

@@ -1,25 +1,25 @@
 from __future__ import annotations
-import copy
+
 import logging
-from typing import Any, List, Text, Dict, Type, Union, Tuple, Optional
+from typing import Any, Dict, List, Optional, Text, Tuple, Type, Union
 
-from rasa.engine.graph import GraphComponent, ExecutionContext
-from rasa.engine.recipes.default_recipe import DefaultV1Recipe
-from rasa.engine.storage.resource import Resource
-from rasa.engine.storage.storage import ModelStorage
-from rasa.shared.constants import DEFAULT_NLU_FALLBACK_INTENT_NAME
 from rasa.core.constants import (
-    DEFAULT_NLU_FALLBACK_THRESHOLD,
     DEFAULT_NLU_FALLBACK_AMBIGUITY_THRESHOLD,
+    DEFAULT_NLU_FALLBACK_THRESHOLD,
 )
+from rasa.engine.graph import ExecutionContext, GraphComponent
+from rasa.engine.recipes.default_recipe import DefaultV1Recipe
+from rasa.engine.storage.resource import Resource
+from rasa.engine.storage.storage import ModelStorage
 from rasa.nlu.classifiers.classifier import IntentClassifier
-from rasa.shared.nlu.training_data.message import Message
+from rasa.shared.constants import DEFAULT_NLU_FALLBACK_INTENT_NAME
 from rasa.shared.nlu.constants import (
     INTENT,
     INTENT_NAME_KEY,
     INTENT_RANKING_KEY,
     PREDICTED_CONFIDENCE_KEY,
 )
+from rasa.shared.nlu.training_data.message import Message
 
 THRESHOLD_KEY = "threshold"
 AMBIGUITY_THRESHOLD_KEY = "ambiguity_threshold"
@@ -185,8 +185,6 @@ def undo_fallback_prediction(prediction: Dict[Text, Any]) -> Dict[Text, Any]:
     if len(intent_ranking) < 2:
         return prediction
 
-    prediction = copy.deepcopy(prediction)
     prediction[INTENT] = intent_ranking[1]
     prediction[INTENT_RANKING_KEY] = prediction[INTENT_RANKING_KEY][1:]
-
     return prediction

rasa/nlu/classifiers/keyword_intent_classifier.py

@@ -1,23 +1,24 @@
 from __future__ import annotations
+
 import logging
 import re
-from typing import Any, Dict, Optional, Text, List
+from typing import Any, Dict, List, Optional, Text
 
-from rasa.engine.graph import GraphComponent, ExecutionContext
+import rasa.shared.utils.io
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.recipes.default_recipe import DefaultV1Recipe
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
-from rasa.shared.constants import DOCS_URL_COMPONENTS
 from rasa.nlu.classifiers.classifier import IntentClassifier
+from rasa.shared.constants import DOCS_URL_COMPONENTS
 from rasa.shared.nlu.constants import (
     INTENT,
-    TEXT,
     INTENT_NAME_KEY,
     PREDICTED_CONFIDENCE_KEY,
+    TEXT,
 )
-import rasa.shared.utils.io
-from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.shared.nlu.training_data.message import Message
+from rasa.shared.nlu.training_data.training_data import TrainingData
 
 logger = logging.getLogger(__name__)
 

rasa/nlu/classifiers/logistic_regression_classifier.py

@@ -1,7 +1,7 @@
-from typing import Any, Text, Dict, List, Type, Tuple
+from typing import Any, Dict, List, Text, Tuple, Type
 
 import structlog
-from scipy.sparse import hstack, vstack, csr_matrix
+from scipy.sparse import csr_matrix, hstack, vstack
 from sklearn.exceptions import NotFittedError
 from sklearn.linear_model import LogisticRegression
 from sklearn.utils.validation import check_is_fitted
@@ -13,7 +13,7 @@ from rasa.engine.storage.storage import ModelStorage
 from rasa.nlu.classifiers import LABEL_RANKING_LENGTH
 from rasa.nlu.classifiers.classifier import IntentClassifier
 from rasa.nlu.featurizers.featurizer import Featurizer
-from rasa.shared.nlu.constants import TEXT, INTENT
+from rasa.shared.nlu.constants import INTENT, TEXT
 from rasa.shared.nlu.training_data.message import Message
 from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.utils.tensorflow.constants import RANKING_LENGTH

rasa/nlu/classifiers/mitie_intent_classifier.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
+
 import logging
-from rasa.nlu.featurizers.featurizer import Featurizer
 import typing
 from typing import Any, Dict, List, Optional, Text, Type
 
@@ -9,11 +9,12 @@ from rasa.engine.recipes.default_recipe import DefaultV1Recipe
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
 from rasa.nlu.classifiers.classifier import IntentClassifier
-from rasa.nlu.utils.mitie_utils import MitieModel, MitieNLP
 from rasa.nlu.constants import TOKENS_NAMES
-from rasa.shared.nlu.constants import TEXT, INTENT
-from rasa.shared.nlu.training_data.training_data import TrainingData
+from rasa.nlu.featurizers.featurizer import Featurizer
+from rasa.nlu.utils.mitie_utils import MitieModel, MitieNLP
+from rasa.shared.nlu.constants import INTENT, TEXT
 from rasa.shared.nlu.training_data.message import Message
+from rasa.shared.nlu.training_data.training_data import TrainingData
 
 if typing.TYPE_CHECKING:
     import mitie

rasa/nlu/classifiers/regex_message_handler.py

@@ -1,8 +1,9 @@
 from __future__ import annotations
+
 import logging
-from typing import Any, Dict, Optional, Text, List
+from typing import Any, Dict, List, Optional, Text
 
-from rasa.engine.graph import GraphComponent, ExecutionContext
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.recipes.default_recipe import DefaultV1Recipe
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage

rasa/nlu/classifiers/sklearn_intent_classifier.py

@@ -8,7 +8,7 @@ from typing import Any, Dict, List, Optional, Text, Tuple, Type
 import numpy as np
 
 import rasa.shared.utils.io
-from rasa.engine.graph import GraphComponent, ExecutionContext
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.recipes.default_recipe import DefaultV1Recipe
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
@@ -291,8 +291,8 @@ class SklearnIntentClassifier(GraphComponent, IntentClassifier):
         **kwargs: Any,
     ) -> SklearnIntentClassifier:
         """Loads trained component (see parent class for full docstring)."""
-        from sklearn.preprocessing import LabelEncoder
         import skops.io as sio
+        from sklearn.preprocessing import LabelEncoder
 
         try:
             with model_storage.read_from(resource) as model_dir:

rasa/nlu/convert.py

@@ -1,10 +1,10 @@
 import os
 from typing import Text, Union
 
-from rasa.shared.nlu.training_data.formats.rasa_yaml import RasaYAMLWriter
-from rasa.shared.utils.cli import print_error
 import rasa.shared.nlu.training_data.loading
 from rasa.nlu.utils import write_to_file
+from rasa.shared.nlu.training_data.formats.rasa_yaml import RasaYAMLWriter
+from rasa.shared.utils.cli import print_error
 
 
 def convert_training_data(

rasa/nlu/emulators/dialogflow.py

@@ -2,16 +2,16 @@ import uuid
 from collections import defaultdict
 from typing import Any, Dict, Text
 
+from rasa.nlu.emulators.emulator import Emulator
 from rasa.shared.nlu.constants import (
-    INTENT_NAME_KEY,
     ENTITIES,
     ENTITY_ATTRIBUTE_TYPE,
     ENTITY_ATTRIBUTE_VALUE,
     INTENT,
-    TEXT,
+    INTENT_NAME_KEY,
     PREDICTED_CONFIDENCE_KEY,
+    TEXT,
 )
-from rasa.nlu.emulators.emulator import Emulator
 
 
 class DialogflowEmulator(Emulator):

rasa/nlu/emulators/luis.py

@@ -3,17 +3,17 @@ from typing import Any, Dict, Text
 from rasa.nlu.emulators.emulator import Emulator
 from rasa.shared.nlu.constants import (
     ENTITIES,
-    ENTITY_ATTRIBUTE_TYPE,
+    ENTITY_ATTRIBUTE_END,
     ENTITY_ATTRIBUTE_ROLE,
-    ENTITY_ATTRIBUTE_VALUE,
     ENTITY_ATTRIBUTE_START,
-    ENTITY_ATTRIBUTE_END,
+    ENTITY_ATTRIBUTE_TYPE,
+    ENTITY_ATTRIBUTE_VALUE,
     EXTRACTOR,
-    INTENT_RANKING_KEY,
-    TEXT,
     INTENT,
     INTENT_NAME_KEY,
+    INTENT_RANKING_KEY,
     PREDICTED_CONFIDENCE_KEY,
+    TEXT,
 )
 
 

rasa/nlu/emulators/no_emulator.py

@@ -1,4 +1,5 @@
 from typing import Any, Dict, Text
+
 from rasa.nlu.emulators.emulator import Emulator
 
 

rasa/nlu/emulators/wit.py

@@ -4,13 +4,13 @@ from typing import Any, Dict, Text
 from rasa.nlu.emulators.emulator import Emulator
 from rasa.shared.nlu.constants import (
     ENTITIES,
-    ENTITY_ATTRIBUTE_TYPE,
-    ENTITY_ATTRIBUTE_ROLE,
-    ENTITY_ATTRIBUTE_VALUE,
     ENTITY_ATTRIBUTE_END,
+    ENTITY_ATTRIBUTE_ROLE,
     ENTITY_ATTRIBUTE_START,
-    TEXT,
+    ENTITY_ATTRIBUTE_TYPE,
+    ENTITY_ATTRIBUTE_VALUE,
     INTENT,
+    TEXT,
 )
 
 

rasa/nlu/extractors/crf_entity_extractor.py

@@ -1,38 +1,38 @@
 from __future__ import annotations
 
-from collections import OrderedDict
-from enum import Enum
 import logging
 import typing
-from typing import Any, Dict, List, Optional, Text, Tuple, Callable, Type
+from collections import OrderedDict
+from enum import Enum
+from typing import Any, Callable, Dict, List, Optional, Text, Tuple, Type
 
 import numpy as np
 
 import rasa.nlu.utils.bilou_utils as bilou_utils
 import rasa.shared.utils.io
 import rasa.utils.train_utils
-from rasa.engine.graph import GraphComponent, ExecutionContext
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.recipes.default_recipe import DefaultV1Recipe
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
+from rasa.nlu.constants import TOKENS_NAMES
+from rasa.nlu.extractors.extractor import EntityExtractorMixin
 from rasa.nlu.test import determine_token_labels
 from rasa.nlu.tokenizers.spacy_tokenizer import POS_TAG_KEY
-from rasa.nlu.extractors.extractor import EntityExtractorMixin
 from rasa.nlu.tokenizers.tokenizer import Token, Tokenizer
-from rasa.shared.nlu.training_data.training_data import TrainingData
-from rasa.shared.nlu.training_data.message import Message
-from rasa.nlu.constants import TOKENS_NAMES
+from rasa.shared.constants import DOCS_URL_COMPONENTS
 from rasa.shared.nlu.constants import (
-    TEXT,
     ENTITIES,
-    ENTITY_ATTRIBUTE_TYPE,
     ENTITY_ATTRIBUTE_GROUP,
     ENTITY_ATTRIBUTE_ROLE,
+    ENTITY_ATTRIBUTE_TYPE,
     NO_ENTITY_TAG,
     SPLIT_ENTITIES_BY_COMMA,
     SPLIT_ENTITIES_BY_COMMA_DEFAULT_VALUE,
+    TEXT,
 )
-from rasa.shared.constants import DOCS_URL_COMPONENTS
+from rasa.shared.nlu.training_data.message import Message
+from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.utils.tensorflow.constants import BILOU_FLAG, FEATURIZERS
 
 logger = logging.getLogger(__name__)

rasa/nlu/extractors/duckling_entity_extractor.py

@@ -1,22 +1,23 @@
 from __future__ import annotations
-import time
+
 import json
 import logging
 import os
+import time
+from typing import Any, Dict, List, Optional, Text
+
 import requests
-from typing import Any, List, Optional, Text, Dict
 
+import rasa.shared.utils.io
 import rasa.utils.endpoints as endpoints_utils
-from rasa.engine.graph import GraphComponent, ExecutionContext
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.recipes.default_recipe import DefaultV1Recipe
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
+from rasa.nlu.extractors.extractor import EntityExtractorMixin
 from rasa.shared.constants import DOCS_URL_COMPONENTS
 from rasa.shared.nlu.constants import ENTITIES, TEXT
-from rasa.nlu.extractors.extractor import EntityExtractorMixin
 from rasa.shared.nlu.training_data.message import Message
-import rasa.shared.utils.io
-
 
 logger = logging.getLogger(__name__)
 

rasa/nlu/extractors/entity_synonyms.py

@@ -1,20 +1,21 @@
 from __future__ import annotations
+
+import logging
 import os
 from typing import Any, Dict, List, Optional, Text
-import logging
 
-from rasa.engine.graph import GraphComponent, ExecutionContext
+import rasa.shared.utils.io
+import rasa.utils.io
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.recipes.default_recipe import DefaultV1Recipe
+from rasa.engine.storage.resource import Resource
+from rasa.engine.storage.storage import ModelStorage
+from rasa.nlu.extractors.extractor import EntityExtractorMixin
+from rasa.nlu.utils import write_json_to_file
 from rasa.shared.constants import DOCS_URL_TRAINING_DATA
 from rasa.shared.nlu.constants import ENTITIES, TEXT
-from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.shared.nlu.training_data.message import Message
-from rasa.nlu.utils import write_json_to_file
-from rasa.nlu.extractors.extractor import EntityExtractorMixin
-import rasa.utils.io
-import rasa.shared.utils.io
-from rasa.engine.storage.resource import Resource
-from rasa.engine.storage.storage import ModelStorage
+from rasa.shared.nlu.training_data.training_data import TrainingData
 
 logger = logging.getLogger(__name__)
 

rasa/nlu/extractors/extractor.py

@@ -1,33 +1,33 @@
 import abc
-from typing import Any, Dict, List, NamedTuple, Text, Tuple, Optional
+from typing import Any, Dict, List, NamedTuple, Optional, Text, Tuple
 
 import rasa.shared.utils.io
-from rasa.shared.constants import DOCS_URL_TRAINING_DATA_NLU
-from rasa.shared.nlu.training_data.training_data import TrainingData
-from rasa.shared.nlu.training_data.message import Message
-from rasa.nlu.tokenizers.tokenizer import Token
+import rasa.utils.train_utils
 from rasa.nlu.constants import (
-    TOKENS_NAMES,
-    ENTITY_ATTRIBUTE_CONFIDENCE_TYPE,
-    ENTITY_ATTRIBUTE_CONFIDENCE_ROLE,
     ENTITY_ATTRIBUTE_CONFIDENCE_GROUP,
+    ENTITY_ATTRIBUTE_CONFIDENCE_ROLE,
+    ENTITY_ATTRIBUTE_CONFIDENCE_TYPE,
+    TOKENS_NAMES,
 )
+from rasa.nlu.tokenizers.tokenizer import Token
+from rasa.shared.constants import DOCS_URL_TRAINING_DATA_NLU
 from rasa.shared.nlu.constants import (
-    TEXT,
-    INTENT,
     ENTITIES,
-    ENTITY_ATTRIBUTE_VALUE,
-    ENTITY_ATTRIBUTE_START,
     ENTITY_ATTRIBUTE_END,
-    EXTRACTOR,
-    ENTITY_ATTRIBUTE_TYPE,
     ENTITY_ATTRIBUTE_GROUP,
     ENTITY_ATTRIBUTE_ROLE,
+    ENTITY_ATTRIBUTE_START,
+    ENTITY_ATTRIBUTE_TYPE,
+    ENTITY_ATTRIBUTE_VALUE,
+    EXTRACTOR,
+    INTENT,
     NO_ENTITY_TAG,
-    SPLIT_ENTITIES_BY_COMMA,
     SINGLE_ENTITY_ALLOWED_INTERLEAVING_CHARSET,
+    SPLIT_ENTITIES_BY_COMMA,
+    TEXT,
 )
-import rasa.utils.train_utils
+from rasa.shared.nlu.training_data.message import Message
+from rasa.shared.nlu.training_data.training_data import TrainingData
 
 
 class EntityTagSpec(NamedTuple):

rasa/nlu/extractors/mitie_entity_extractor.py

@@ -1,29 +1,30 @@
 from __future__ import annotations
+
 import logging
-from rasa.nlu.tokenizers.tokenizer import Tokenizer
 import typing
 from typing import Any, Dict, List, Optional, Text, Type
 
-from rasa.engine.graph import GraphComponent, ExecutionContext
+import rasa.shared.utils.io
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.recipes.default_recipe import DefaultV1Recipe
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
 from rasa.nlu.constants import TOKENS_NAMES
+from rasa.nlu.extractors.extractor import EntityExtractorMixin
+from rasa.nlu.tokenizers.tokenizer import Tokenizer
+from rasa.nlu.utils.mitie_utils import MitieModel, MitieNLP
+from rasa.shared.exceptions import InvalidConfigException
 from rasa.shared.nlu.constants import (
+    ENTITIES,
     ENTITY_ATTRIBUTE_CONFIDENCE,
-    ENTITY_ATTRIBUTE_START,
     ENTITY_ATTRIBUTE_END,
+    ENTITY_ATTRIBUTE_START,
     ENTITY_ATTRIBUTE_TYPE,
     ENTITY_ATTRIBUTE_VALUE,
     TEXT,
-    ENTITIES,
 )
-from rasa.nlu.utils.mitie_utils import MitieModel, MitieNLP
-from rasa.nlu.extractors.extractor import EntityExtractorMixin
-from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.shared.nlu.training_data.message import Message
-import rasa.shared.utils.io
-from rasa.shared.exceptions import InvalidConfigException
+from rasa.shared.nlu.training_data.training_data import TrainingData
 
 logger = logging.getLogger(__name__)
 

rasa/nlu/extractors/regex_entity_extractor.py

@@ -1,25 +1,26 @@
 from __future__ import annotations
+
 import logging
 import re
 from typing import Any, Dict, List, Optional, Text
 
-from rasa.engine.graph import GraphComponent, ExecutionContext
+import rasa.nlu.utils.pattern_utils as pattern_utils
+import rasa.shared.utils.io
+from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.recipes.default_recipe import DefaultV1Recipe
-from rasa.engine.storage.storage import ModelStorage
 from rasa.engine.storage.resource import Resource
-import rasa.shared.utils.io
-import rasa.nlu.utils.pattern_utils as pattern_utils
-from rasa.shared.nlu.training_data.training_data import TrainingData
-from rasa.shared.nlu.training_data.message import Message
+from rasa.engine.storage.storage import ModelStorage
+from rasa.nlu.extractors.extractor import EntityExtractorMixin
 from rasa.shared.nlu.constants import (
     ENTITIES,
-    ENTITY_ATTRIBUTE_VALUE,
-    ENTITY_ATTRIBUTE_START,
     ENTITY_ATTRIBUTE_END,
-    TEXT,
+    ENTITY_ATTRIBUTE_START,
     ENTITY_ATTRIBUTE_TYPE,
+    ENTITY_ATTRIBUTE_VALUE,
+    TEXT,
 )
-from rasa.nlu.extractors.extractor import EntityExtractorMixin
+from rasa.shared.nlu.training_data.message import Message
+from rasa.shared.nlu.training_data.training_data import TrainingData
 
 logger = logging.getLogger(__name__)
 

rasa/nlu/extractors/spacy_entity_extractor.py

@@ -5,9 +5,9 @@ from rasa.engine.graph import ExecutionContext, GraphComponent
 from rasa.engine.recipes.default_recipe import DefaultV1Recipe
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
-from rasa.shared.nlu.constants import ENTITIES, TEXT
-from rasa.nlu.utils.spacy_utils import SpacyModel, SpacyNLP
 from rasa.nlu.extractors.extractor import EntityExtractorMixin
+from rasa.nlu.utils.spacy_utils import SpacyModel, SpacyNLP
+from rasa.shared.nlu.constants import ENTITIES, TEXT
 from rasa.shared.nlu.training_data.message import Message
 
 if typing.TYPE_CHECKING: