From 2c3feb8d73b37721c259230b7d58ff66d188278b Mon Sep 17 00:00:00 2001 From: Shicheng Xu Date: Mon, 30 Jun 2025 13:53:37 -0700 Subject: [PATCH] TBD. PiperOrigin-RevId: 777704014 --- fiddle/_src/absl_flags/flags.py | 7 +++++++ 1 file changed, 7 insertions(+) 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,