Internal change. purely make the presubmit happy. to be deleted once the diffbase is removed.

PiperOrigin-RevId: 631644170

Author: Fiddle-Config Team

fiddle/_src/experimental/auto_config.py

@@ -601,6 +601,12 @@ def build_model():
   )
 
 
+def inlined_partial(
+    auto_config_fn: AutoConfig, *args, **kwargs
+) -> functools.partial:
+  return functools.partial(auto_config_fn, *args, **kwargs)
+
+
 @dataclasses.dataclass(frozen=True)
 class ConfigTypes:
   config_cls: Type[config.Config] = config.Config
@@ -759,6 +765,19 @@ def auto_config_call_handler(fn_or_cls, *args, **kwargs):
               for name, arg in kwargs.items()
           },
       )
+    elif fn_or_cls is inlined_partial:
+      auto_config_fn, partial_args = args[0], args[1:]
+      if not isinstance(auto_config_fn, AutoConfig):
+        raise ValueError(
+            'inlined_partial should only be applied to auto_config functions.'
+        )
+      cfg = auto_config_fn.as_buildable(*partial_args, **kwargs)
+      if not isinstance(cfg, config.Config):
+        raise ValueError(
+            'inlined_partial should only be applied to auto_config functions'
+            ' that create a single top-level Config.'
+        )
+      return cast_lib.cast(partial_cls, cfg)
 
     if fn_or_cls is exempt:
       return fn_or_cls(*args, **kwargs)

fiddle/_src/experimental/auto_config_test.py

@@ -318,6 +318,24 @@ def test_fn_config_2():
     self.assertEqual(expected_config, test_fn_config_1.as_buildable())
     self.assertEqual(expected_config, test_fn_config_2.as_buildable())
 
+  def test_inlined_partial(self):
+    expected_config = fdl.Partial(FrozenSampleClass, arg1=1, arg2=2)
+
+    @auto_config.auto_config
+    def make_frozen_sample_class(arg1):
+      return FrozenSampleClass(arg1, 2)
+
+    @auto_config.auto_config
+    def test_config():
+      return auto_config.inlined_partial(make_frozen_sample_class, arg1=1)
+
+    self.assertEqual(expected_config, test_config.as_buildable())
+
+    python_path_output = test_config()
+    self.assertIsInstance(python_path_output, functools.partial)
+    self.assertEqual(python_path_output.func, make_frozen_sample_class)
+    self.assertEqual(python_path_output.keywords, {'arg1': 1})
+
   def test_create_config_with_args(self):
     expected_config = fdl.Config(
         FrozenSampleClass, 'positional', arg2='default'

fiddle/experimental/auto_config.py

@@ -30,6 +30,7 @@
 from fiddle._src.experimental.auto_config import ConfigTypes
 from fiddle._src.experimental.auto_config import exempt
 from fiddle._src.experimental.auto_config import inline
+from fiddle._src.experimental.auto_config import inlined_partial
 from fiddle._src.experimental.auto_config import is_auto_config
 from fiddle._src.experimental.auto_config import UnsupportedLanguageConstructError
 from fiddle._src.experimental.with_tags import with_tags