From 351c1f5f383add598198e16e1753ac7684fce3fd Mon Sep 17 00:00:00 2001
From: Fiddle-Config Team <fiddle-config-team@google.com>
Date: Tue, 4 Feb 2025 11:10:58 -0800
Subject: [PATCH] Test only.

PiperOrigin-RevId: 723158497
---
 fiddle/_src/absl_flags/utils.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/fiddle/_src/absl_flags/utils.py b/fiddle/_src/absl_flags/utils.py
index 40e36bb0..6c62bb40 100644
--- a/fiddle/_src/absl_flags/utils.py
+++ b/fiddle/_src/absl_flags/utils.py
@@ -274,7 +274,12 @@ def resolve_function_reference(
     except ModuleNotFoundError as e:
       raise ValueError(f'{failure_msg_prefix} {function_name!r}: {e}') from e
   else:
-    available_names = module_reflection.find_base_config_like_things(module)
+    if mode == ImportDottedNameDebugContext.BASE_CONFIG:
+      available_names = module_reflection.find_base_config_like_things(module)
+    elif mode == ImportDottedNameDebugContext.FIDDLER:
+      available_names = module_reflection.find_fiddler_like_things(module)
+    else:
+      raise ValueError(f'Unknown mode: {mode}')
     raise ValueError(
         f'{failure_msg_prefix} {function_name!r}: Could not resolve reference '
         f'to named function, available names: {", ".join(available_names)}.'