code: def lr(self):
      if self.n_steps <= self.n_warmup_steps:
          return self.init_lr + self.warmup_k * self.n_steps
      else:
          return self.max_lr
  
imports: import numpy
  from torch.optim import Optimizer
  from typing import Tuple
label_desc: Use image node_id 2 for calling the ConstantScheduler obj's underlying member method code with example usage: obj.lr() and returns: unknown, self