code: def __init__(self):
      super().__init__()
      self.layer = torch.nn.Linear(32, 2)
  
file_path: DeepSpeed/tests/lightning/test_simple.py
imports: import torch
  from pytorch_lightning import LightningModule, Trainer
  from pytorch_lightning.strategies import DeepSpeedStrategy
  from torch.utils.data import DataLoader, Dataset
label_desc: Use image node_id 1 to create a new BoringModel object from inherited base classes: LightningModule with example: obj = BoringModel()