diff --git a/fiddle/_src/absl_flags/flags.py b/fiddle/_src/absl_flags/flags.py index f0fca53b..97e28c97 100644 --- a/fiddle/_src/absl_flags/flags.py +++ b/fiddle/_src/absl_flags/flags.py @@ -132,6 +132,8 @@ def __init__( # Due to the lazy evaluation of `value`, this list is needed to keep # track of the remaining `directives`. self._remaining_directives = [] + # Save all arguments if needed for forwarding. + self._all_arguments = [] super().__init__(*args, **kwargs) def _initial_config(self, expression: str): @@ -176,6 +178,7 @@ def _apply_fiddler(self, cfg: config.Buildable, expression: str): return new_cfg if new_cfg is not None else cfg def parse(self, arguments): + self._all_arguments.append(arguments) new_parsed = self._parse(arguments) self._remaining_directives.extend(new_parsed) self.present += len(new_parsed) @@ -259,6 +262,10 @@ def value(self): def value(self, value): self._value = value + @property + def all_arguments(self): + return self._all_arguments + def DEFINE_fiddle_config( # pylint: disable=invalid-name name: str,