diff --git a/pipelinewise/cli/utils.py b/pipelinewise/cli/utils.py index 9015b4dcf..67b3abe90 100644 --- a/pipelinewise/cli/utils.py +++ b/pipelinewise/cli/utils.py @@ -196,7 +196,7 @@ def load_yaml(yaml_file, vault_secret=None): try: if is_encrypted_file(stream): file_data = stream.read() - data = yaml.load(vault.decrypt(file_data, None)) + data = yaml.safe_load(vault.decrypt(file_data, None)) else: loader = AnsibleLoader(stream, None, vault.secrets) try: diff --git a/setup.py b/setup.py index e9b06b31e..8ee90bce6 100644 --- a/setup.py +++ b/setup.py @@ -19,7 +19,7 @@ install_requires=[ 'argparse==1.4.0', 'tabulate==0.8.9', - 'PyYAML==5.4.1', + 'PyYAML==6.0', 'ansible==4.7.0', 'Jinja2==3.0.2', 'joblib==1.1.0',