Internal change

PiperOrigin-RevId: 447079920

official/modeling/hyperparams/params_dict.py

@@ -41,15 +41,15 @@ _PARAM_RE = re.compile(
 
 _CONST_VALUE_RE = re.compile(r'(\d.*|-\d.*|None)')
 
-# Yaml loader with an implicit resolver to parse float decimal and exponential
+# Yaml LOADER with an implicit resolver to parse float decimal and exponential
 # format. The regular experission parse the following cases:
 # 1- Decimal number with an optional exponential term.
 # 2- Integer number with an exponential term.
 # 3- Decimal number with an optional exponential term.
 # 4- Decimal number.
 
-LOADER = yaml.SafeLoader
-LOADER.add_implicit_resolver(
+_LOADER = yaml.SafeLoader
+_LOADER.add_implicit_resolver(
     'tag:yaml.org,2002:float',
     re.compile(r'''
     ^(?:[-+]?(?:[0-9][0-9_]*)\\.[0-9_]*(?:[eE][-+]?[0-9]+)?
@@ -332,7 +332,7 @@ class ParamsDict(object):
 def read_yaml_to_params_dict(file_path: str):
   """Reads a YAML file to a ParamsDict."""
   with tf.io.gfile.GFile(file_path, 'r') as f:
-    params_dict = yaml.load(f, Loader=LOADER)
+    params_dict = yaml.load(f, Loader=_LOADER)
     return ParamsDict(params_dict)
 
 
@@ -453,7 +453,7 @@ def override_params_dict(params, dict_or_string_or_yaml_file, is_strict):
           nested_csv_str_to_json_str(dict_or_string_or_yaml_file))
     except ValueError:
       pass
-    params_dict = yaml.load(dict_or_string_or_yaml_file, Loader=LOADER)
+    params_dict = yaml.load(dict_or_string_or_yaml_file, Loader=_LOADER)
     if isinstance(params_dict, dict):
       params.override(params_dict, is_strict)
     else: