Mark incremental training experimental (#7543)

* add the marking * add missing import Co-authored-by: Tobias Wochinger <[email protected]>
RasaHQ · Dec 14, 2020 · 235b5f9 · 235b5f9
1 parent 977de59
commit 235b5f9
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 5 deletions.
diff --git a/docs/docs/command-line-interface.mdx b/docs/docs/command-line-interface.mdx
@@ -87,6 +87,14 @@ The following arguments can be used to configure the training process:
 
 ### Incremental training
 
+:::caution
+This feature is experimental.
+We introduce experimental features to get feedback from our community, so we encourage you to try it out!
+However, the functionality might be changed or removed in the future.
+If you have feedback (positive or negative) please share it with us on the [Rasa Forum](https://forum.rasa.com).
+
+:::
+
 In order to improve the performance of an assistant, it's helpful to practice [CDD](./conversation-driven-development.mdx)
 and add new training examples based on how your users have talked to your assistant. You can use `rasa train --finetune`
 to initialize the pipeline with an already trained model and further finetune it on the

diff --git a/rasa/train.py b/rasa/train.py
@@ -3,32 +3,29 @@
 import tempfile
 from contextlib import ExitStack
 from typing import (
-    Any,
     Text,
     NamedTuple,
     Tuple,
     Optional,
     List,
     Union,
     Dict,
-    TYPE_CHECKING,
 )
 
 import rasa.core.interpreter
 from rasa.shared.nlu.interpreter import NaturalLanguageInterpreter
 from rasa.shared.importers.importer import TrainingDataImporter
 from rasa import model, telemetry
-from rasa.model import Fingerprint, FingerprintComparisonResult
+from rasa.model import FingerprintComparisonResult
 from rasa.shared.core.domain import Domain
 from rasa.nlu.model import Interpreter
 import rasa.utils.common
+import rasa.shared.utils.common
 from rasa.utils.common import TempDirectoryPath
 
 from rasa.shared.utils.cli import (
     print_success,
     print_warning,
-    print_error,
-    print_color,
 )
 import rasa.shared.exceptions
 import rasa.shared.utils.io
@@ -551,6 +548,9 @@ async def _train_core_with_validated_data(
         )
 
         if model_to_finetune:
+            rasa.shared.utils.common.mark_as_experimental_feature(
+                "Incremental Training feature"
+            )
             model_to_finetune = await _core_model_for_finetuning(
                 model_to_finetune,
                 file_importer=file_importer,
@@ -756,6 +756,9 @@ async def _train_nlu_with_validated_data(
         )
 
         if model_to_finetune:
+            rasa.shared.utils.common.mark_as_experimental_feature(
+                "Incremental Training feature"
+            )
             model_to_finetune = await _nlu_model_for_finetuning(
                 model_to_finetune,
                 file_importer,