code: def __init__(self):
      super().__init__()
      self.conv = nn.Conv2d(3, 8, kernel_size=10)
      self.relu = nn.ReLU()
      self.dropout = nn.Dropout()
      self.linear = nn.Linear(8, 1)
      self.sigmoid = nn.Sigmoid()
  
imports: import math
  import unittest
  from unittest.mock import Mock
  import numpy
  import pytest
  import torch
  import torchmetrics
  from torch import nn
  from torch.utils.data import Dataset, DataLoader
  from baal.modelwrapper import ModelWrapper, mc_inference
  from baal.utils.metrics import ClassificationReport
label_desc: Use image node_id 1 to create a new SimpleModel object from inherited base classes: nn with example: obj = SimpleModel()