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

rasa/engine/recipes/default_components.py

@@ -1,17 +1,21 @@
 from rasa.core.policies.enterprise_search_policy import EnterpriseSearchPolicy
 from rasa.core.policies.flow_policy import FlowPolicy
+from rasa.core.policies.intentless_policy import IntentlessPolicy
+from rasa.core.policies.memoization import AugmentedMemoizationPolicy, MemoizationPolicy
+from rasa.core.policies.rule_policy import RulePolicy
+from rasa.core.policies.ted_policy import TEDPolicy
+from rasa.core.policies.unexpected_intent_policy import UnexpecTEDIntentPolicy
 from rasa.dialogue_understanding.coexistence.intent_based_router import (
     IntentBasedRouter,
 )
 from rasa.dialogue_understanding.coexistence.llm_based_router import LLMBasedRouter
-from rasa.core.policies.intentless_policy import IntentlessPolicy
+from rasa.dialogue_understanding.generator import (
+    LLMCommandGenerator,
+)
 from rasa.dialogue_understanding.generator.nlu_command_adapter import NLUCommandAdapter
 from rasa.nlu.classifiers.diet_classifier import DIETClassifier
 from rasa.nlu.classifiers.fallback_classifier import FallbackClassifier
 from rasa.nlu.classifiers.keyword_intent_classifier import KeywordIntentClassifier
-from rasa.dialogue_understanding.generator import (
-    LLMCommandGenerator,
-)
 from rasa.nlu.classifiers.logistic_regression_classifier import (
     LogisticRegressionClassifier,
 )
@@ -21,18 +25,18 @@ from rasa.nlu.extractors.crf_entity_extractor import CRFEntityExtractor
 from rasa.nlu.extractors.duckling_entity_extractor import DucklingEntityExtractor
 from rasa.nlu.extractors.entity_synonyms import EntitySynonymMapper
 from rasa.nlu.extractors.mitie_entity_extractor import MitieEntityExtractor
-from rasa.nlu.extractors.spacy_entity_extractor import SpacyEntityExtractor
 from rasa.nlu.extractors.regex_entity_extractor import RegexEntityExtractor
-from rasa.nlu.featurizers.sparse_featurizer.lexical_syntactic_featurizer import (
-    LexicalSyntacticFeaturizer,
-)
+from rasa.nlu.extractors.spacy_entity_extractor import SpacyEntityExtractor
 from rasa.nlu.featurizers.dense_featurizer.convert_featurizer import ConveRTFeaturizer
+from rasa.nlu.featurizers.dense_featurizer.lm_featurizer import LanguageModelFeaturizer
 from rasa.nlu.featurizers.dense_featurizer.mitie_featurizer import MitieFeaturizer
 from rasa.nlu.featurizers.dense_featurizer.spacy_featurizer import SpacyFeaturizer
 from rasa.nlu.featurizers.sparse_featurizer.count_vectors_featurizer import (
     CountVectorsFeaturizer,
 )
-from rasa.nlu.featurizers.dense_featurizer.lm_featurizer import LanguageModelFeaturizer
+from rasa.nlu.featurizers.sparse_featurizer.lexical_syntactic_featurizer import (
+    LexicalSyntacticFeaturizer,
+)
 from rasa.nlu.featurizers.sparse_featurizer.regex_featurizer import RegexFeaturizer
 from rasa.nlu.selectors.response_selector import ResponseSelector
 from rasa.nlu.tokenizers.jieba_tokenizer import JiebaTokenizer
@@ -42,12 +46,6 @@ from rasa.nlu.tokenizers.whitespace_tokenizer import WhitespaceTokenizer
 from rasa.nlu.utils.mitie_utils import MitieNLP
 from rasa.nlu.utils.spacy_utils import SpacyNLP
 
-
-from rasa.core.policies.ted_policy import TEDPolicy
-from rasa.core.policies.memoization import MemoizationPolicy, AugmentedMemoizationPolicy
-from rasa.core.policies.rule_policy import RulePolicy
-from rasa.core.policies.unexpected_intent_policy import UnexpecTEDIntentPolicy
-
 DEFAULT_COMPONENTS = [
     # Message Classifiers
     DIETClassifier,

rasa/engine/recipes/default_recipe.py

@@ -1,65 +1,72 @@
 from __future__ import annotations
 
 import copy
+import dataclasses
 import enum
 import logging
 import math
 from enum import Enum
-from typing import Dict, Text, Any, Tuple, Type, Optional, List, Callable, Set, Union
-
-import dataclasses
+from typing import Any, Callable, Dict, List, Optional, Set, Text, Tuple, Type, Union
 
+import rasa.shared.constants
 from rasa.core.featurizers.precomputation import (
-    CoreFeaturizationInputConverter,
     CoreFeaturizationCollector,
+    CoreFeaturizationInputConverter,
 )
-from rasa.graph_components.providers.flows_provider import FlowsProvider
+from rasa.core.policies.ensemble import DefaultPolicyPredictionEnsemble
 from rasa.dialogue_understanding.processor.command_processor_component import (
     CommandProcessorComponent,
 )
-from rasa.shared.exceptions import FileNotFoundException
-from rasa.core.policies.ensemble import DefaultPolicyPredictionEnsemble
-
-from rasa.engine.graph import (
-    GraphSchema,
-    GraphComponent,
-    SchemaNode,
-    GraphModelConfiguration,
-)
 from rasa.engine.constants import (
+    PLACEHOLDER_ENDPOINTS,
     PLACEHOLDER_IMPORTER,
     PLACEHOLDER_MESSAGE,
     PLACEHOLDER_TRACKER,
-    PLACEHOLDER_ENDPOINTS,
+)
+from rasa.engine.graph import (
+    GraphComponent,
+    GraphModelConfiguration,
+    GraphSchema,
+    SchemaNode,
 )
 from rasa.engine.recipes.recipe import Recipe
 from rasa.engine.storage.resource import Resource
 from rasa.graph_components.converters.nlu_message_converter import NLUMessageConverter
-from rasa.graph_components.providers.domain_provider import DomainProvider
-from rasa.graph_components.providers.forms_provider import FormsProvider
-from rasa.graph_components.providers.responses_provider import ResponsesProvider
 from rasa.graph_components.providers.domain_for_core_training_provider import (
     DomainForCoreTrainingProvider,
 )
+from rasa.graph_components.providers.domain_provider import DomainProvider
+from rasa.graph_components.providers.flows_provider import FlowsProvider
+from rasa.graph_components.providers.forms_provider import FormsProvider
 from rasa.graph_components.providers.nlu_training_data_provider import (
     NLUTrainingDataProvider,
 )
+from rasa.graph_components.providers.responses_provider import ResponsesProvider
 from rasa.graph_components.providers.rule_only_provider import RuleOnlyDataProvider
 from rasa.graph_components.providers.story_graph_provider import StoryGraphProvider
 from rasa.graph_components.providers.training_tracker_provider import (
     TrainingTrackerProvider,
 )
-import rasa.shared.constants
-from rasa.shared.exceptions import RasaException, InvalidConfigException
-from rasa.shared.constants import ASSISTANT_ID_KEY
+from rasa.shared.constants import (
+    ASSISTANT_ID_KEY,
+    CONFIG_ADDITIONAL_LANGUAGES_KEY,
+    CONFIG_LANGUAGE_KEY,
+    CONFIG_NAME_KEY,
+    CONFIG_PIPELINE_KEY,
+    CONFIG_POLICIES_KEY,
+)
 from rasa.shared.data import TrainingType
-from rasa.shared.utils.yaml import read_config_file
-
-from rasa.utils.tensorflow.constants import EPOCHS
+from rasa.shared.exceptions import (
+    FileNotFoundException,
+    InvalidConfigException,
+    RasaException,
+)
 from rasa.shared.utils.common import (
     class_from_module_path,
     transform_collection_to_sentence,
 )
+from rasa.shared.utils.yaml import read_config_file
+from rasa.utils.tensorflow.constants import EPOCHS
 
 logger = logging.getLogger(__name__)
 
@@ -67,13 +74,13 @@ logger = logging.getLogger(__name__)
 DEFAULT_PREDICT_KWARGS = dict(constructor_name="load", eager=True, is_target=False)
 
 COMMENTS_FOR_KEYS = {
-    "pipeline": (
+    CONFIG_PIPELINE_KEY: (
         f"# # No configuration for the NLU pipeline was provided. The following "
         f"default pipeline was used to train your model.\n"
         f"# # If you'd like to customize it, uncomment and adjust the pipeline.\n"
         f"# # See {rasa.shared.constants.DOCS_URL_PIPELINE} for more information.\n"
     ),
-    "policies": (
+    CONFIG_POLICIES_KEY: (
         f"# # No configuration for policies was provided. The following default "
         f"policies were used to train your model.\n"
         f"# # If you'd like to customize them, uncomment and adjust the policies.\n"
@@ -82,6 +89,10 @@ COMMENTS_FOR_KEYS = {
 }
 
 
+GRAPH_NODE_RUN_PREFIX = "run_"
+GRAPH_NODE_TRAIN_PREFIX = "train_"
+
+
 class DefaultV1RecipeRegisterException(RasaException):
     """If you register a class which is not of type `GraphComponent`."""
 
@@ -194,10 +205,12 @@ class DefaultV1Recipe(Recipe):
     ) -> GraphModelConfiguration:
         """Converts the default config to graphs (see interface for full docstring)."""
         self._use_core = (
-            bool(config.get("policies")) and not training_type == TrainingType.NLU
+            bool(config.get(CONFIG_POLICIES_KEY))
+            and not training_type == TrainingType.NLU
         )
         self._use_nlu = (
-            bool(config.get("pipeline")) and not training_type == TrainingType.CORE
+            bool(config.get(CONFIG_PIPELINE_KEY))
+            and not training_type == TrainingType.CORE
         )
 
         if not self._use_nlu and training_type == TrainingType.NLU:
@@ -232,9 +245,10 @@ class DefaultV1Recipe(Recipe):
             predict_schema=GraphSchema(predict_nodes),
             training_type=training_type,
             assistant_id=config.get(ASSISTANT_ID_KEY),
-            language=config.get("language"),
+            language=config.get(CONFIG_LANGUAGE_KEY),
             core_target=core_target,
-            nlu_target=f"run_{RegexMessageHandler.__name__}",
+            nlu_target=f"{GRAPH_NODE_RUN_PREFIX}{RegexMessageHandler.__name__}",
+            additional_languages=config.get(CONFIG_ADDITIONAL_LANGUAGES_KEY),
         )
 
     def _create_train_nodes(
@@ -317,7 +331,7 @@ class DefaultV1Recipe(Recipe):
             constructor_name="create",
             fn="provide",
             config={
-                "language": train_config.get("language"),
+                "language": train_config.get(CONFIG_LANGUAGE_KEY),
                 "persist": persist_nlu_data,
             },
             is_target=persist_nlu_data,
@@ -327,8 +341,8 @@ class DefaultV1Recipe(Recipe):
         last_run_node = "nlu_training_data_provider"
         preprocessors: List[Text] = []
 
-        for idx, config in enumerate(train_config["pipeline"]):
-            component_name = config.pop("name")
+        for idx, config in enumerate(train_config[CONFIG_PIPELINE_KEY]):
+            component_name = config.pop(CONFIG_NAME_KEY)
             component = self._from_registry(component_name)
             component_name = f"{component_name}{idx}"
 
@@ -451,7 +465,7 @@ class DefaultV1Recipe(Recipe):
         needs.update(self._get_model_provider_needs(train_nodes, component))
         needs["training_data"] = last_run_node
 
-        train_node_name = f"train_{component_name}"
+        train_node_name = f"{GRAPH_NODE_TRAIN_PREFIX}{component_name}"
         train_nodes[train_node_name] = SchemaNode(
             needs=needs,
             uses=component,
@@ -469,10 +483,10 @@ class DefaultV1Recipe(Recipe):
         component_config: Dict[Text, Any],
     ) -> Dict[Text, Any]:
         from rasa.nlu.classifiers.mitie_intent_classifier import MitieIntentClassifier
-        from rasa.nlu.extractors.mitie_entity_extractor import MitieEntityExtractor
         from rasa.nlu.classifiers.sklearn_intent_classifier import (
             SklearnIntentClassifier,
         )
+        from rasa.nlu.extractors.mitie_entity_extractor import MitieEntityExtractor
 
         cli_args_mapping: Dict[Type[GraphComponent], List[Text]] = {
             MitieIntentClassifier: ["num_threads"],
@@ -520,7 +534,7 @@ class DefaultV1Recipe(Recipe):
 
         needs["training_data"] = last_run_node
 
-        node_name = f"run_{component_name}"
+        node_name = f"{GRAPH_NODE_RUN_PREFIX}{component_name}"
         train_nodes[node_name] = SchemaNode(
             needs=needs,
             uses=component_class,
@@ -627,8 +641,8 @@ class DefaultV1Recipe(Recipe):
         )
 
         policy_with_end_to_end_support_used = False
-        for idx, config in enumerate(train_config["policies"]):
-            component_name = config.pop("name")
+        for idx, config in enumerate(train_config[CONFIG_POLICIES_KEY]):
+            component_name = config.pop(CONFIG_NAME_KEY)
             component = self._from_registry(component_name)
 
             extra_config_from_cli = self._extra_config_from_cli(
@@ -647,7 +661,7 @@ class DefaultV1Recipe(Recipe):
                 needs["precomputations"] = "end_to_end_features_provider"
             # during core training we use a stripped down version of the domain
             needs["domain"] = "domain_for_core_training_provider"
-            train_nodes[f"train_{component_name}{idx}"] = SchemaNode(
+            train_nodes[f"{GRAPH_NODE_TRAIN_PREFIX}{component_name}{idx}"] = SchemaNode(
                 needs=needs,
                 uses=component.clazz,
                 constructor_name="load" if self._is_finetuning else "create",
@@ -722,7 +736,9 @@ class DefaultV1Recipe(Recipe):
         if self._use_core:
             domain_needs["domain"] = "domain_provider"
 
-        regex_handler_node_name = f"run_{RegexMessageHandler.__name__}"
+        regex_handler_node_name = (
+            f"{GRAPH_NODE_RUN_PREFIX}{RegexMessageHandler.__name__}"
+        )
         predict_nodes[regex_handler_node_name] = SchemaNode(
             **DEFAULT_PREDICT_KWARGS,
             needs={"messages": last_run_nlu_node, **domain_needs},
@@ -762,8 +778,8 @@ class DefaultV1Recipe(Recipe):
             resource=Resource("domain_provider"),
         )
 
-        for idx, config in enumerate(predict_config["pipeline"]):
-            component_name = config.pop("name")
+        for idx, config in enumerate(predict_config[CONFIG_PIPELINE_KEY]):
+            component_name = config.pop(CONFIG_NAME_KEY)
             component = self._from_registry(component_name)
             component_name = f"{component_name}{idx}"
             if self.ComponentType.MODEL_LOADER in component.types:
@@ -830,10 +846,10 @@ class DefaultV1Recipe(Recipe):
         item_config: Dict[Text, Any],
         from_resource: bool = False,
     ) -> Text:
-        train_node_name = f"run_{node_name}"
+        train_node_name = f"{GRAPH_NODE_RUN_PREFIX}{node_name}"
         resource = None
         if from_resource:
-            train_node_name = f"train_{node_name}"
+            train_node_name = f"{GRAPH_NODE_TRAIN_PREFIX}{node_name}"
             resource = Resource(train_node_name)
 
         return self._add_nlu_predict_node(
@@ -852,7 +868,7 @@ class DefaultV1Recipe(Recipe):
         component_name: Text,
         last_run_node: Text,
     ) -> Text:
-        node_name = f"run_{component_name}"
+        node_name = f"{GRAPH_NODE_RUN_PREFIX}{component_name}"
 
         needs = self._get_needs_from_args(node.uses, "process")
         needs.update(self._get_model_provider_needs(predict_nodes, node.uses))
@@ -919,12 +935,12 @@ class DefaultV1Recipe(Recipe):
         rule_policy_resource = None
         policies: List[Text] = []
 
-        for idx, config in enumerate(predict_config["policies"]):
-            component_name = config.pop("name")
+        for idx, config in enumerate(predict_config[CONFIG_POLICIES_KEY]):
+            component_name = config.pop(CONFIG_NAME_KEY)
             component = self._from_registry(component_name)
 
-            train_node_name = f"train_{component_name}{idx}"
-            node_name = f"run_{component_name}{idx}"
+            train_node_name = f"{GRAPH_NODE_TRAIN_PREFIX}{component_name}{idx}"
+            node_name = f"{GRAPH_NODE_RUN_PREFIX}{component_name}{idx}"
 
             from rasa.core.policies.rule_policy import RulePolicy
 

rasa/engine/recipes/graph_recipe.py

@@ -1,16 +1,18 @@
 import logging
+from typing import Any, Dict, Text, Tuple
 
+from rasa.engine.graph import GraphModelConfiguration, GraphSchema
 from rasa.engine.recipes.recipe import Recipe
-from rasa.engine.graph import GraphModelConfiguration
-from rasa.shared.constants import DOCS_URL_GRAPH_RECIPE, ASSISTANT_ID_KEY
+from rasa.shared.constants import (
+    ASSISTANT_ID_KEY,
+    CONFIG_ADDITIONAL_LANGUAGES_KEY,
+    CONFIG_LANGUAGE_KEY,
+    DOCS_URL_GRAPH_RECIPE,
+)
 from rasa.shared.data import TrainingType
 from rasa.shared.exceptions import InvalidConfigException
 from rasa.shared.utils.common import mark_as_experimental_feature
 from rasa.shared.utils.io import raise_warning
-from rasa.engine.graph import GraphSchema
-
-from typing import Dict, Text, Any, Tuple
-
 
 logger = logging.getLogger(__name__)
 
@@ -72,7 +74,8 @@ class GraphV1Recipe(Recipe):
             predict_schema=GraphSchema.from_dict(config.get("predict_schema")),
             training_type=training_type,
             assistant_id=config.get(ASSISTANT_ID_KEY),
-            language=config.get("language"),
+            language=config.get(CONFIG_LANGUAGE_KEY),
             core_target=core_target,
             nlu_target=nlu_target,
+            additional_languages=config.get(CONFIG_ADDITIONAL_LANGUAGES_KEY),
         )

rasa/engine/recipes/recipe.py

@@ -1,12 +1,12 @@
 from __future__ import annotations
 
 import abc
-from typing import Text, Dict, Any, Optional, Tuple, Set
+from typing import Any, Dict, Optional, Set, Text, Tuple
 
 import rasa.shared.utils.io
 from rasa.engine.graph import GraphModelConfiguration
-from rasa.shared.exceptions import RasaException
 from rasa.shared.data import TrainingType
+from rasa.shared.exceptions import RasaException
 
 
 class InvalidRecipeException(RasaException):

rasa/engine/runner/dask.py

@@ -1,11 +1,11 @@
 from __future__ import annotations
 
+import asyncio
 import logging
 from typing import Any, Dict, List, Optional, Text
 
-import dask.local
 import dask.core
-import asyncio
+import dask.local
 
 from rasa.engine.exceptions import GraphRunError
 from rasa.engine.graph import ExecutionContext, GraphNode, GraphNodeHook, GraphSchema

rasa/engine/runner/interface.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 from abc import ABC, abstractmethod
 from typing import Any, Dict, List, Optional, Text
 

rasa/engine/storage/local_model_storage.py

@@ -8,17 +8,18 @@ import uuid
 from contextlib import contextmanager
 from datetime import datetime
 from pathlib import Path
-from tarsafe import TarSafe
 from typing import Generator, Optional, Text, Tuple, Union
 
-import rasa.utils.common
+from tarsafe import TarSafe
+
+import rasa.model
 import rasa.shared.utils.io
-from rasa.engine.storage.storage import ModelMetadata, ModelStorage
+import rasa.utils.common
 from rasa.engine.graph import GraphModelConfiguration
 from rasa.engine.storage.resource import Resource
+from rasa.engine.storage.storage import ModelMetadata, ModelStorage
 from rasa.exceptions import UnsupportedModelVersionError
 from rasa.shared.core.domain import Domain
-import rasa.model
 
 logger = logging.getLogger(__name__)
 
@@ -239,6 +240,7 @@ class LocalModelStorage(ModelStorage):
             training_type=model_configuration.training_type,
             project_fingerprint=rasa.model.project_fingerprint(),
             language=model_configuration.language,
+            additional_languages=model_configuration.additional_languages,
             core_target=model_configuration.core_target,
             nlu_target=model_configuration.nlu_target,
         )

rasa/engine/storage/resource.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
+
 import logging
 import typing
+import uuid
 from dataclasses import dataclass, field
 from pathlib import Path
 from typing import Text
-import uuid
 
 import rasa.utils.common
 import rasa.utils.io

rasa/engine/storage/storage.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 import abc
 import logging
 import typing
@@ -6,17 +7,18 @@ from contextlib import contextmanager
 from dataclasses import dataclass
 from datetime import datetime
 from pathlib import Path
-from typing import Tuple, Union, Text, Generator, Dict, Any, Optional
+from typing import Any, Dict, Generator, List, Optional, Text, Tuple, Union
+
 from packaging import version
 
 from rasa.constants import MINIMUM_COMPATIBLE_VERSION
-from rasa.exceptions import UnsupportedModelVersionError
 from rasa.engine.storage.resource import Resource
+from rasa.exceptions import UnsupportedModelVersionError
 from rasa.shared.core.domain import Domain
 from rasa.shared.data import TrainingType
 
 if typing.TYPE_CHECKING:
-    from rasa.engine.graph import GraphSchema, GraphModelConfiguration
+    from rasa.engine.graph import GraphModelConfiguration, GraphSchema
 
 logger = logging.getLogger(__name__)
 
@@ -140,6 +142,7 @@ class ModelMetadata:
     core_target: Optional[Text]
     nlu_target: Text
     language: Optional[Text]
+    additional_languages: Optional[List[Text]]
     training_type: TrainingType = TrainingType.BOTH
 
     def __post_init__(self) -> None:
@@ -169,6 +172,7 @@ class ModelMetadata:
             "core_target": self.core_target,
             "nlu_target": self.nlu_target,
             "language": self.language,
+            "additional_languages": self.additional_languages,
         }
 
     @classmethod
@@ -196,4 +200,5 @@ class ModelMetadata:
             core_target=serialized["core_target"],
             nlu_target=serialized["nlu_target"],
             language=serialized["language"],
+            additional_languages=serialized.get("additional_languages"),
         )

rasa/engine/training/components.py

@@ -1,7 +1,8 @@
 from __future__ import annotations
-from typing import Any, Dict, Optional, Text, Type
+
 import dataclasses
 import uuid
+from typing import Any, Dict, Optional, Text, Type
 
 from rasa.engine.caching import Cacheable, TrainingCache
 from rasa.engine.graph import ExecutionContext, GraphComponent, SchemaNode

rasa/engine/training/fingerprinting.py

@@ -1,10 +1,12 @@
 import inspect
 import logging
 from typing import Any, Dict, Text, Type
-from typing_extensions import Protocol, runtime_checkable
+
 import importlib_metadata
-import rasa.utils.common
+from typing_extensions import Protocol, runtime_checkable
+
 import rasa.shared.utils.io
+import rasa.utils.common
 from rasa.engine.graph import GraphComponent
 
 logger = logging.getLogger(__name__)

rasa/engine/training/graph_trainer.py

@@ -4,16 +4,16 @@ from pathlib import Path
 from typing import Any, Dict, Text, Type, Union
 
 from rasa.engine.caching import TrainingCache
-from rasa.engine.graph import ExecutionContext, GraphSchema, GraphModelConfiguration
 from rasa.engine.constants import PLACEHOLDER_IMPORTER
+from rasa.engine.graph import ExecutionContext, GraphModelConfiguration, GraphSchema
 from rasa.engine.runner.interface import GraphRunner
-from rasa.engine.storage.storage import ModelStorage, ModelMetadata
+from rasa.engine.storage.storage import ModelMetadata, ModelStorage
 from rasa.engine.training.components import (
-    PrecomputedValueProvider,
     FingerprintComponent,
     FingerprintStatus,
+    PrecomputedValueProvider,
 )
-from rasa.engine.training.hooks import TrainingHook, LoggingHook
+from rasa.engine.training.hooks import LoggingHook, TrainingHook
 from rasa.shared.importers.importer import TrainingDataImporter
 
 logger = logging.getLogger(__name__)

rasa/engine/training/hooks.py

@@ -1,12 +1,12 @@
 import logging
 from typing import Any, Dict, Text, Type
 
+import rasa.shared.utils.io
 from rasa.engine.caching import TrainingCache
 from rasa.engine.graph import ExecutionContext, GraphNodeHook, GraphSchema, SchemaNode
 from rasa.engine.storage.storage import ModelStorage
-from rasa.engine.training.components import PrecomputedValueProvider
-import rasa.shared.utils.io
 from rasa.engine.training import fingerprinting
+from rasa.engine.training.components import PrecomputedValueProvider
 
 logger = logging.getLogger(__name__)