Merge pull request #1129 from jstriebel/patch-1

Allow to run evaluate script from other code via main function.
RasaHQ · Jun 1, 2018 · 35a53a1 · 35a53a1
2 parents 271cebf + 2fc65b7
commit 35a53a1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/rasa_nlu/evaluate.py b/rasa_nlu/evaluate.py
@@ -685,7 +685,7 @@ def return_entity_results(results, dataset_name):
             return_results(result, dataset_name)
 
 
-if __name__ == '__main__':  # pragma: no cover
+def main():
     parser = create_argument_parser()
     cmdline_args = parser.parse_args()
 
@@ -723,3 +723,7 @@ def return_entity_results(results, dataset_name):
         run_evaluation(cmdline_args.data, cmdline_args.model)
 
     logger.info("Finished evaluation")
+
+
+if __name__ == '__main__':  # pragma: no cover
+    main()