Absorb signatrue change to PyYAML

cf-platform-eng · Jul 29, 2022 · c541d50 · c541d50
1 parent c990ab5
commit c541d50
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tile_generator/config_unittest.py b/tile_generator/config_unittest.py
@@ -156,8 +156,8 @@ def deep_comparer(self, expected, given, path):
 
     else:
       try:
-        loaded_expected = yaml.load(expected)
-        loaded_given = yaml.load(given)
+        loaded_expected = yaml.load(expected, yaml.Loader)
+        loaded_given = yaml.load(given, yaml.Loader)
       except:
         loaded_expected = expected
         loaded_given = given
@@ -184,10 +184,10 @@ def test_metadata_diff(self, mock_read_history, mock_base64_img):
     template.render(test_path + '/test_metadata_generated_output.yml', 'tile/metadata.yml', cfg)
 
     with open(test_path + '/test_metadata_generated_output.yml', 'r') as f:
-      generated_output = yaml.load(f)
+      generated_output = yaml.load(f, yaml.Loader)
 
     with open(test_path + '/test_metadata_expected_output.yml', 'r') as f:
-      expected_output = yaml.load(f)
+      expected_output = yaml.load(f, yaml.Loader)
 
     self.deep_comparer(expected_output, generated_output, '[%s]')