move log messages after training

RasaHQ · Jan 3, 2019 · e537ce8 · e537ce8
1 parent b995751
commit e537ce8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 10 deletions.
diff --git a/examples/concertbot/train.py b/examples/concertbot/train.py
@@ -8,11 +8,6 @@ def train_dialogue(domain_file='domain.yml',
                    stories_file='data/stories.md',
                    model_path='models/dialogue',
                    policy_config='policy_config.yml'):
-    utils.configure_colored_logging(loglevel="INFO")
-    logger.info("This example does not include NLU data."
-                "Please specify the desired intent with a preceding '/', e.g."
-                "'/greet' .")
-
     return train.train_dialogue_model(domain_file=domain_file,
                                       stories_file=stories_file,
                                       output_path=model_path,
@@ -24,3 +19,8 @@ def train_dialogue(domain_file='domain.yml',
 
 if __name__ == '__main__':
     train_dialogue()
+
+    utils.configure_colored_logging(loglevel="INFO")
+    logger.info("This example does not include NLU data."
+                "Please specify the desired intent with a preceding '/', e.g."
+                "'/greet' .")
diff --git a/examples/concertbot/train_interactive.py b/examples/concertbot/train_interactive.py
@@ -7,11 +7,6 @@
 
 
 def train_agent():
-    utils.configure_colored_logging(loglevel="INFO")
-    logger.info("This example does not include NLU data."
-                "Please specify the desired intent with a preceding '/', e.g."
-                "'/greet' .")
-
     return train.train_dialogue_model(domain_file="domain.yml",
                                       stories_file="data/stories.md",
                                       output_path="models/dialogue",
@@ -22,4 +17,7 @@ def train_agent():
 if __name__ == '__main__':
     utils.configure_colored_logging(loglevel="INFO")
     agent = train_agent()
+    logger.info("This example does not include NLU data."
+                "Please specify the desired intent with a preceding '/', e.g."
+                "'/greet' .")
     interactive.run_interactive_learning(agent)