fix: language not associated with right model

botfront · Aug 19, 2020 · bd8b860 · bd8b860
1 parent 14b58f0
commit bd8b860
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/rasa/train.py b/rasa/train.py
@@ -492,7 +492,7 @@ async def _train_nlu_with_validated_data(
         config = await file_importer.get_nlu_config(retrain_nlu)
         for lang in config:
             if config[lang]:
-                print_color("Start training {} NLU model ...".format(lang), color=bcolors.OKBLUE)
+                print_color("Start training <{}> NLU model ...".format(lang), color=bcolors.OKBLUE)
                 _, models[lang], _ = await rasa.nlu.train(
                     config[lang],
                     file_importer,

diff --git a/rasa_addons/importers/botfront.py b/rasa_addons/importers/botfront.py
@@ -39,7 +39,7 @@ def __init__(
                 config = io_utils.read_config_file(file)
                 lang = config["language"]
                 self.core_config = {"policies": config["policies"]}
-                self.nlu_config[lang] = {"pipeline": config["pipeline"], "data": lang}
+                self.nlu_config[lang] = {"pipeline": config["pipeline"], "language": lang}
 
     def path_for_nlu_lang(self, lang) -> Text:
         return [x for x in self._nlu_files if "{}.md".format(lang) in x]