code: def __init__(self, training_module: LightningModule):
      super().__init__()
      self.training_module = training_module
  
imports: from __future__ import annotations
  import warnings
  from typing import Any, Iterable, List, cast, TYPE_CHECKING
  import torch.optim
  import torch.nn
  from torch.optim import Optimizer
  import nni.nas.nn.pytorch
  from nni.nas.evaluator.pytorch import LightningModule, Trainer
  from nni.mutable import Sample
  from .supermodule.base import BaseSuperNetModule
label_desc: Use image node_id 2 to create a new BaseOneShotLightningModule object from inherited base classes: LightningModule with example: obj = BaseOneShotLightningModule(training_module)