use yaml spec 1.2 so that yes does not get parsed to boolean value …

…`True`
RasaHQ · Nov 19, 2018 · 7fd27ab · 7fd27ab
1 parent 155f82e
commit 7fd27ab
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 3 deletions.
diff --git a/data/examples/rasa/demo-rasa.json b/data/examples/rasa/demo-rasa.json
@@ -22,12 +22,12 @@
     ],
     "common_examples": [
       {
-        "text": "hey", 
+        "text": "hey",
         "intent": "greet", 
         "entities": []
       }, 
       {
-        "text": "howdy", 
+        "text": "howdy",
         "intent": "greet", 
         "entities": []
       }, 

diff --git a/rasa_nlu/utils/__init__.py b/rasa_nlu/utils/__init__.py
@@ -251,7 +251,7 @@ def read_yaml(content):
     replace_environment_variables()
 
     yaml_parser = yaml.YAML(typ="safe")
-    yaml_parser.version = "1.1"
+    yaml_parser.version = "1.2"
     yaml_parser.unicode_supplementary = True
 
     return yaml_parser.load(content)

diff --git a/tests/base/test_utils.py b/tests/base/test_utils.py
@@ -222,3 +222,17 @@ def test_emojis_in_tmp_file():
 
     assert actual["data"][0] == "one 😁"
     assert actual["data"][1] == "two £"
+
+
+def test_bool_str():
+    test_data = """
+    one: "yes"
+    two: "true"
+    three: "True"
+    """
+
+    actual = utils.read_yaml(test_data)
+
+    assert actual["one"] == "yes"
+    assert actual["two"] == "true"
+    assert actual["three"] == "True"