updating Error message propagation in module_configdict. Some exception were not correctly captured

PiperOrigin-RevId: 886772537

Author: The kauldron Authors

kauldron/konfig/flags_utils.py

@@ -121,7 +121,7 @@ def value(self):
     _value = self._value  # pylint: disable=invalid-name
     assert _value is not None, "None for _value is not supported."
     if isinstance(_value, module_configdict.ModuleConfigDict):
-      return _value.module_config
+      return _value.get_module_config()
     else:
       # absl.testing.flagsaver.save_flag_values copies the value into ._value
       # and then later calls the setter with the original value. This is why

kauldron/konfig/module_configdict.py

@@ -118,8 +118,7 @@ def config_args(self) -> Any:
     else:
       return self.cli_str_arg
 
-  @functools.cached_property
-  def module_config(self) -> konfig.ConfigDict:
+  def get_module_config(self) -> konfig.ConfigDict:
     """Initialize config from config file.
 
     Returns:
@@ -195,4 +194,4 @@ def get_config_from_module(
 ) -> konfig.ConfigDict:
   """Get config from module and cli_str_arg."""
   modulecfg = ModuleConfigDict(module=module, cli_str_arg=cli_str_arg)
-  return modulecfg.module_config
+  return modulecfg.get_module_config()

kauldron/utils/colab.py

@@ -79,7 +79,7 @@ def iter_sweep_configs(
         kontext.set_by_path(module_cfg, k, v)
       # Only for visualization.
 
-      cfg = module_cfg.module_config
+      cfg = module_cfg.get_module_config()
 
       sweep_cfg_overwrites = konfig.ConfigDict(sweep_kwargs)
       print(f'Work-unit {i+1}:', flush=True)