Merge 2e7aa3f into 9b09956

etsy · Apr 28, 2019 · 8c86d61 · 8c86d61
2 parents 9b09956 + 2e7aa3f
commit 8c86d61
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/boundary_layer/registry/registry.py b/boundary_layer/registry/registry.py
@@ -173,7 +173,7 @@ def node_configs(self):
     def load_from_file(self, filename):
         item = None
         with open(filename) as _in:
-            item = yaml.load(_in)
+            item = yaml.safe_load(_in)
 
         logger.debug('validating item %s against schema %s',
                      item, self.spec_schema_cls.__name__)

diff --git a/test/test_graph.py b/test/test_graph.py
@@ -31,7 +31,7 @@ def imports(self):
     type = NodeTypes.OPERATOR
 
     def __init__(self, item):
-        mock_config = yaml.load(MOCK_NODE_CONFIG_YAML)
+        mock_config = yaml.safe_load(MOCK_NODE_CONFIG_YAML)
         super(MockNode, self).__init__(mock_config, item)