No public description

PiperOrigin-RevId: 568424866

Author: baek-jinoo

official/modeling/hyperparams/params_dict_test.py

@@ -220,7 +220,7 @@ def test_save_params_dict_to_yaml(self):
     params_dict.save_params_dict_to_yaml(params, output_yaml_file)
 
     with tf.io.gfile.GFile(output_yaml_file, 'r') as f:
-      params_d = yaml.load(f)
+      params_d = yaml.load(f, Loader=yaml.Loader)
       self.assertEqual(params.a, params_d['a'])
       self.assertEqual(params.b, params_d['b'])
       self.assertEqual(params.c.c1, params_d['c']['c1'])
@@ -364,7 +364,7 @@ def test_basic_csv_str_load(self):
     csv_str = 'a=1,b=2,c=3'
     expected_output = {'a': 1, 'b': 2, 'c': 3}
     converted_csv_str = params_dict.nested_csv_str_to_json_str(csv_str)
-    converted_dict = yaml.load(converted_csv_str)
+    converted_dict = yaml.load(converted_csv_str, Loader=yaml.Loader)
     self.assertDictEqual(converted_dict, expected_output)
 
   def test_basic_nested_csv_str_to_json_str(self):
@@ -377,7 +377,7 @@ def test_basic_nested_csv_str_load(self):
     csv_str = 'a=1,b.b1=2,c.c1=3'
     expected_output = {'a': 1, 'b': {'b1': 2}, 'c': {'c1': 3}}
     converted_csv_str = params_dict.nested_csv_str_to_json_str(csv_str)
-    converted_dict = yaml.load(converted_csv_str)
+    converted_dict = yaml.load(converted_csv_str, Loader=yaml.Loader)
     self.assertDictEqual(converted_dict, expected_output)
 
   def test_complex_nested_csv_str_to_json_str(self):
@@ -390,7 +390,7 @@ def test_complex_nested_csv_str_load(self):
     csv_str = 'a.aa.aaa.aaaaa.a=1,a.a=2'
     expected_output = {'a': {'aa': {'aaa': {'aaaaa': {'a': 1}}}, 'a': 2}}
     converted_csv_str = params_dict.nested_csv_str_to_json_str(csv_str)
-    converted_dict = yaml.load(converted_csv_str)
+    converted_dict = yaml.load(converted_csv_str, Loader=yaml.Loader)
     self.assertDictEqual(converted_dict, expected_output)
 
   def test_int_array_param_nested_csv_str_to_json_str(self):
@@ -413,7 +413,7 @@ def test_incomplete_array_param_nested_csv_str_to_json_str(self):
   def test_csv_str_load_supported_datatypes(self):
     csv_str = 'a=1,b=2.,c=[1,2,3],d=\'hello, there\',e=\"Hi.\"'
     converted_csv_str = params_dict.nested_csv_str_to_json_str(csv_str)
-    converted_dict = yaml.load(converted_csv_str)
+    converted_dict = yaml.load(converted_csv_str, Loader=yaml.Loader)
     self.assertEqual(converted_dict['a'], 1)
     self.assertEqual(converted_dict['b'], 2.)
     self.assertEqual(converted_dict['c'], [1, 2, 3])