code: def get_lr(self) -> Tensor:
      if self.epoch_counter.numpy()[0] not in self.milestones:
          return self.optimizer.lr
      return self.optimizer.lr * self.gamma
  
imports: import math
  from typing import List
  from tinygrad.nn.optim import Optimizer
  from tinygrad.tensor import Tensor
label_desc: Use image node_id 2 for calling the MultiStepLR obj's underlying member method code with example usage: obj.get_lr() and returns: unknown, self