Use yaml safe_load function

gunthercox · Apr 29, 2020 · 8cd8f5f · 8cd8f5f
1 parent 80e2272
commit 8cd8f5f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chatterbot/corpus.py b/chatterbot/corpus.py
@@ -35,7 +35,7 @@ def read_corpus(file_name):
     Read and return the data from a corpus json file.
     """
     with io.open(file_name, encoding='utf-8') as data_file:
-        return yaml.load(data_file)
+        return yaml.safe_load(data_file)
 
 
 def list_corpus_files(dotted_path):