Internal

PiperOrigin-RevId: 720607527
google · Jan 28, 2025 · c42ce44 · c42ce44
1 parent 21ed6f3
commit c42ce44
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/grain/_src/core/config.py b/grain/_src/core/config.py
@@ -155,10 +155,11 @@ def __getattr__(self, name: str) -> Any:
     flag_name = f"grain_{name}"
     if any(f.name == flag_name for f in _GRAIN_FLAGS):
       value = getattr(flags.FLAGS, flag_name)
-      enrolled = int(
-          bool(value) if value is None or isinstance(value, str) else value
-      )
-      _grain_experiment_metric.Set(enrolled, flag_name)
+      if isinstance(value, int):
+        int_value = value
+      else:
+        int_value = int(value != flags.FLAGS[flag_name].default)
+      _grain_experiment_metric.Set(int_value, flag_name)
       return value
     raise ValueError(f"Unrecognized config option: {name}")