code: def __init__(self):
      super().__init__()
  
      self.layer_1 = torch.nn.Linear(28 * 28, 128)
      self.layer_2 = torch.nn.Linear(128, 10)
      self.batch_size = 8
  
imports: import json
  from glob import glob
  import os
  import torch
  from torch.nn import functional
  from torch import nn
  from pytorch_lightning.core.lightning import LightningModule
  from pytorch_lightning import Trainer
  from torch.optim import Adam
  from torch.utils.data import DataLoader, random_split, Subset
  from torch.utils.data.dataset import Dataset
  from torchvision.datasets import MNIST
  from torchvision import transforms
  import pytest
  import urllib
  from keepsake.pl_callback import KeepsakeCallback
label_desc: Use image node_id 1 to create a new ModelNoValidation object from inherited base classes: LightningModule with example: obj = ModelNoValidation()