Bump version of pyyaml

belvedere-trading · Oct 26, 2021 · 58f3944 · 58f3944
1 parent e2e9524
commit 58f3944
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/required_packages.req b/required_packages.req
@@ -1,3 +1,3 @@
 click>=6.0
-PyYAML>=3.0
+PyYAML>=5.4.1
 voluptuous>=0.10.0
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -5,7 +5,7 @@
 
 from util import VigilanceTestCase, mock_decorator
 
-@mock.patch('yaml.load')
+@mock.patch('yaml.full_load')
 @mock.patch('vigilance.cli.open', create=True)
 class CliTest(VigilanceTestCase):
     @property

diff --git a/vigilance/cli.py b/vigilance/cli.py
@@ -25,7 +25,7 @@ def main(configFile): #pylint: disable=missing-docstring, invalid-name
     The default configuration file if no options are passed is vigilance.yaml within the current working directory.
     """
     try:
-        suites = ConfigurationSchema(yaml.load(configFile))['suites']
+        suites = ConfigurationSchema(yaml.full_load(configFile))['suites']
     except yaml.YAMLError:
         raise ConfigurationParsingError('Could not load configuration file as yaml')
     except Invalid as ex: