Skip to content

Commit

Permalink
Fix flags reload on colab
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 673342710
  • Loading branch information
Conchylicultor authored and The kauldron Authors committed Sep 11, 2024
1 parent 5d08a03 commit f9b78ad
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions kauldron/konfig/flags_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
"""Flags utils."""

from absl import flags
from absl.flags import _helpers
from kauldron.konfig import configdict_base
from ml_collections import config_flags

Expand Down Expand Up @@ -52,4 +53,7 @@ def DEFINE_config_file( # pylint: disable=g-bad-name
flag_values=flag_values,
lock_config=lock_config,
accept_new_attributes=True,
# Pass the module name of the caller, otherwise the flag is assumed to
# be created in this file, which creates duplicated flag errors.
module_name=_helpers.get_calling_module(),
)

0 comments on commit f9b78ad

Please sign in to comment.