Restore --flagfile fix compatibility

PiperOrigin-RevId: 695220238

docs/README.md

@@ -5,4 +5,4 @@ The ML Collections documentation can be found here: https://ml-collections.readt
 # How to build the docs
 1. Install the requirements in ml_collections/docs/requirements.txt.
 2. Ensure `pandoc` is installed.
-3. Run `make html` to locally generate documentation.
+3. Run `make html` to locally generate documentation.

ml_collections/config_flags/config_flags.py

@@ -720,6 +720,12 @@ def __init__(
     self._sys_argv = sys_argv
     super(_ConfigFlag, self).__init__(**kwargs)
 
+  def _GetArgv(self):
+    """Lazily fetches sys.argv and expands any potential --flagfile=..."""
+    argv = sys.argv if self._sys_argv is None else self._sys_argv
+    argv = flags.FLAGS.read_flags_from_files(argv, force_gnu=False)
+    return argv
+
   def _GetOverrides(self, argv):
     """Parses the command line arguments for the overrides."""
     # We use a dict to keep the order of the overrides.
@@ -756,8 +762,7 @@ def _IsConfigSpecified(self, argv):
     return self._FindConfigSpecified(argv) >= 0
 
   def _set_default(self, default):
-    if self._IsConfigSpecified(
-        sys.argv if self._sys_argv is None else self._sys_argv):
+    if self._IsConfigSpecified(self._GetArgv()):
       self.default = default
     else:
       super(_ConfigFlag, self)._set_default(default)  # pytype: disable=attribute-error
@@ -787,8 +792,7 @@ def _parse(self, argument):
     config = super(_ConfigFlag, self)._parse(argument)
 
     # Get list or overrides
-    overrides = self._GetOverrides(
-        sys.argv if self._sys_argv is None else self._sys_argv)
+    overrides = self._GetOverrides(self._GetArgv())
     # Iterate over overridden fields and create valid parsers
     self._override_values = {}
     self._initialize_missing_parent_fields(config, overrides)

ml_collections/config_flags/tests/config_overriding_test.py

@@ -17,6 +17,7 @@
 import copy
 import shlex
 import sys
+import tempfile
 
 from absl import flags
 from absl.testing import absltest
@@ -832,6 +833,19 @@ def testOverridesSerialize(self):
                        serialize_parse('test_config.type_tuple',
                                        values.test_config.type_tuple))
 
+  def testFlagfile(self):
+    config = config_dict.ConfigDict()
+    config.foo = 3
+    config.bar = 4
+
+    with tempfile.NamedTemporaryFile(mode='w', delete=False) as f:
+      f.write('--test_config.foo=7\n')
+      f.flush()
+      f.close()
+      values = _parse_flags(f'./program --flagfile={f.name}', config=config)
+      self.assertEqual(values.test_config.foo, 7)
+      self.assertEqual(values.test_config.bar, 4)
+
 
 def main():
   absltest.main()