code: def get_warmup_and_linear_decay(max_steps, warmup_steps):
      warmup_steps = max(warmup_steps, 1)
      return lambda step: min(
          step / warmup_steps,
          1.0 - (step - warmup_steps) / (max_steps - warmup_steps),
      )
  
label_desc: Use image node_id 1 for calling a global function with example usage: get_warmup_and_linear_decay(max_steps, warmup_steps) and returns: function