def update_run_name_from_quantization_config()

in dags/multipod/configs/maxtext_sweep_gke_config.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

def update_run_name_from_quantization_config(run_name, configs):
  """Updates the run name based on quantization configuration.

  Args:
    run_name: The original run name.
    configs: A dictionary of configurations.

  Returns:
    The updated run name.
  """
  quantization_value = configs.get("M_QUANTIZATION")
  if quantization_value is not None:
    run_name += "_" + (
        "bf16" if quantization_value == "" else quantization_value
    )
  return run_name