Merge pull request #51 from crcornwell/master

specify the loader in yaml.load call
heavenshell · May 4, 2019 · 9c3f6f6 · 9c3f6f6
2 parents b065799 + 3c0fe47
commit 9c3f6f6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sqlalchemy_seed/__init__.py b/sqlalchemy_seed/__init__.py
@@ -14,6 +14,10 @@
 import os
 
 import yaml
+try:
+    from yaml import CLoader as Loader
+except ImportError:
+    from yaml import Loader
 
 __version__ = '0.1.1'
 
@@ -59,7 +63,7 @@ def load_fixture_files(paths, files):
 
             with open(fixture_path, 'r') as f:
                 if file.endswith('.yaml') or file.endswith('.yml'):
-                    data = yaml.load(f)
+                    data = yaml.load(f, Loader=Loader)
                 elif file.endswith('.json'):
                     data = json.loads(f)
                 else: