diff --git a/tests/integration/canary_test/util.py b/tests/integration/canary_test/util.py index 8d7f2aa6..8bab1fb0 100644 --- a/tests/integration/canary_test/util.py +++ b/tests/integration/canary_test/util.py @@ -133,5 +133,5 @@ def load_config(config, dir=''): os.path.dirname(os.path.abspath(__file__)) + dir, config ) with open(config_file, "r") as f: - config = yaml.load(f) + config = yaml.safe_load(f) return config diff --git a/tests/integration/util.py b/tests/integration/util.py index 9c6153cc..bffa171c 100644 --- a/tests/integration/util.py +++ b/tests/integration/util.py @@ -17,5 +17,5 @@ def load_config(config, dir=''): os.path.dirname(os.path.abspath(__file__)) + dir, config ) with open(config_file, "r") as f: - config = yaml.load(f) + config = yaml.safe_load(f) return config diff --git a/tools/deploy/cluster.py b/tools/deploy/cluster.py index ef81158a..5627c1b4 100644 --- a/tools/deploy/cluster.py +++ b/tools/deploy/cluster.py @@ -75,7 +75,7 @@ def load(cfg_file): """ with open(cfg_file, 'r') as f: try: - cfg = yaml.load(f) + cfg = yaml.safe_load(f) except yaml.YAMLError as ex: print 'Failed to unmarshal cluster config %s' % cfg_file raise ex diff --git a/tools/minicluster/main.py b/tools/minicluster/main.py index 71c97810..8b2e2f57 100755 --- a/tools/minicluster/main.py +++ b/tools/minicluster/main.py @@ -57,7 +57,7 @@ def load_config(): os.path.dirname(os.path.abspath(__file__)), "config.yaml" ) with open(config_file, "r") as f: - config = yaml.load(f) + config = yaml.safe_load(f) return config diff --git a/tools/vcluster/config_generator.py b/tools/vcluster/config_generator.py index a8873a8c..f8c69227 100644 --- a/tools/vcluster/config_generator.py +++ b/tools/vcluster/config_generator.py @@ -9,7 +9,7 @@ def load_config(config_file): with open(config_file, 'r') as f: - config = yaml.load(f) + config = yaml.safe_load(f) return config