code: def get_lr(self) -> Tensor:
      return (
          self.epoch_counter < self.total_steps * self.pct_start
      ).where(
          self._annealing_linear(
              self.initial_lr,
              self.max_lr,
              self.epoch_counter / (self.total_steps * self.pct_start),
          ),
          self._annealing_linear(
              self.max_lr,
              self.min_lr,
              (self.epoch_counter - (self.total_steps * self.pct_start))
              / (self.total_steps * (1 - self.pct_start)),
          ),
      )
  
imports: import math
  from typing import List
  from tinygrad.nn.optim import Optimizer
  from tinygrad.tensor import Tensor
label_desc: Use image node_id 3 for calling the OneCycleLR obj's underlying member method code with example usage: obj.get_lr() and returns: unknown