rasa-pro 3.11.5__py3-none-any.whl → 3.12.0__py3-none-any.whl

rasa/model_testing.py

@@ -1,32 +1,31 @@
 import copy
 import logging
 import os
+from pathlib import Path
 from typing import (
-    Text,
-    Dict,
-    Optional,
-    List,
     Any,
+    Dict,
     Iterable,
+    List,
+    Optional,
+    Text,
     Tuple,
     Union,
 )
-from pathlib import Path
 
-from rasa.core.agent import Agent
-from rasa.engine.storage.local_model_storage import LocalModelStorage
+import rasa.model
+import rasa.shared.nlu.training_data.loading
 import rasa.shared.utils.cli
 import rasa.shared.utils.common
 import rasa.shared.utils.io
 import rasa.utils.common
-from rasa.constants import RESULTS_FILE, NUMBER_OF_TRAINING_STORIES_FILE
+from rasa.constants import NUMBER_OF_TRAINING_STORIES_FILE, RESULTS_FILE
+from rasa.core.agent import Agent
+from rasa.engine.storage.local_model_storage import LocalModelStorage
 from rasa.exceptions import ModelNotFound
 from rasa.shared.constants import DEFAULT_RESULTS_PATH
-import rasa.shared.nlu.training_data.loading
 from rasa.shared.data import TrainingType
 from rasa.shared.nlu.training_data.training_data import TrainingData
-import rasa.model
-
 
 logger = logging.getLogger(__name__)
 
@@ -245,10 +244,9 @@ async def compare_nlu_models(
     exclusion_percentages: List[int],
 ) -> None:
     """Trains multiple models, compares them and saves the results."""
-    from rasa.nlu.test import drop_intents_below_freq
+    from rasa.nlu.test import compare_nlu, drop_intents_below_freq
     from rasa.nlu.utils import write_json_to_file
     from rasa.utils.io import create_path
-    from rasa.nlu.test import compare_nlu
 
     test_data = drop_intents_below_freq(test_data, cutoff=5)
 
@@ -313,10 +311,10 @@ async def perform_nlu_cross_validation(
             cross-validation, like number of `disable_plotting`.
     """
     from rasa.nlu.test import (
-        drop_intents_below_freq,
         cross_validate,
-        log_results,
+        drop_intents_below_freq,
         log_entity_results,
+        log_results,
     )
 
     additional_arguments = additional_arguments or {}

rasa/model_training.py

@@ -23,6 +23,7 @@ from rasa.engine.storage.local_model_storage import LocalModelStorage
 from rasa.engine.storage.storage import ModelStorage
 from rasa.engine.training.components import FingerprintStatus
 from rasa.engine.training.graph_trainer import GraphTrainer
+from rasa.shared.constants import CONFIG_RECIPE_KEY
 from rasa.shared.core.domain import Domain
 from rasa.shared.core.events import SlotSet
 from rasa.shared.core.training_data.structures import StoryGraph
@@ -139,6 +140,7 @@ def _check_unresolved_slots(domain: Domain, stories: StoryGraph) -> None:
                 f"whether there is a spelling error."
             ),
         )
+        rasa.shared.utils.common.display_research_study_prompt()
         sys.exit(1)
 
 
@@ -295,6 +297,7 @@ async def _train_graph(
                     f"model within the directory '{output_path}'."
                 ),
             )
+            rasa.shared.utils.common.display_research_study_prompt()
             sys.exit(1)
 
         rasa.shared.utils.common.mark_as_experimental_feature(
@@ -304,7 +307,7 @@ async def _train_graph(
     is_finetuning = model_to_finetune is not None
 
     config = file_importer.get_config()
-    recipe = Recipe.recipe_for_name(config.get("recipe"))
+    recipe = Recipe.recipe_for_name(config.get(CONFIG_RECIPE_KEY))
     config, _missing_keys, _configured_keys = recipe.auto_configure(
         file_importer.get_config_file_for_auto_config(),
         config,
@@ -351,35 +354,32 @@ async def _train_graph(
         model_name = determine_model_name(fixed_model_name, training_type)
         full_model_path = Path(output_path, model_name)
 
-        with telemetry.track_model_training(
-            file_importer, model_type=training_type.model_type
-        ):
-            await trainer.train(
-                model_configuration,
-                file_importer,
-                full_model_path,
-                force_retraining=force_full_training,
-                is_finetuning=is_finetuning,
+        await trainer.train(
+            model_configuration,
+            file_importer,
+            full_model_path,
+            force_retraining=force_full_training,
+            is_finetuning=is_finetuning,
+        )
+        if remote_storage:
+            push_model_to_remote_storage(full_model_path, remote_storage)
+            if not keep_local_model_copy:
+                full_model_path.unlink()
+            structlogger.info(
+                "model_training.train.finished_training",
+                event_info=(
+                    f"Your Rasa model {model_name} is trained "
+                    f"and saved at remote storage provider '{remote_storage}'."
+                ),
             )
-            if remote_storage:
-                push_model_to_remote_storage(full_model_path, remote_storage)
-                if not keep_local_model_copy:
-                    full_model_path.unlink()
-                structlogger.info(
-                    "model_training.train.finished_training",
-                    event_info=(
-                        f"Your Rasa model {model_name} is trained "
-                        f"and saved at remote storage provider '{remote_storage}'."
-                    ),
-                )
-            else:
-                structlogger.info(
-                    "model_training.train.finished_training",
-                    event_info=(
-                        f"Your Rasa model is trained and saved at '{full_model_path}'."
-                    ),
-                )
-
+        else:
+            structlogger.info(
+                "model_training.train.finished_training",
+                event_info=(
+                    f"Your Rasa model is trained and saved at '{full_model_path}'."
+                ),
+            )
+        rasa.shared.utils.common.display_research_study_prompt()
         return TrainingResult(str(full_model_path), 0)
 
 

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,24 +1,25 @@
 from __future__ import annotations
+
 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"

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__)