code: def __init__(
      self, lr_scheduler: torch.optim.lr_scheduler._LRScheduler
  ) -> None:
      self.lr_scheduler = lr_scheduler
  
imports: from typing import Any, Dict
  import torch
  from allennlp.common.checks import ConfigurationError
  from allennlp.common.registrable import Registrable
  from allennlp.training.optimizers import Optimizer
  from allennlp.training.scheduler import Scheduler
  from transformers.optimization import get_constant_schedule, get_constant_schedule_with_warmup, get_cosine_schedule_with_warmup, get_cosine_with_hard_restarts_schedule_with_warmup
label_desc: Use image node_id 1 to create a new _PyTorchLearningRateSchedulerWrapper object from inherited base classes: LearningRateScheduler with example: obj = _PyTorchLearningRateSchedulerWrapper(lr_scheduler)