def METHOD_NAME(current_step):
    current_step += steps_shift
    if current_step < warmup_steps:
        return float(current_step) / float(max(1, warmup_steps))
    return max(
        1e-7,
        float(total_training_steps - current_step)
        / float(max(1, total_training_steps - warmup_steps)),
    )