code: def __init__(self):
      super(TestModule, self).__init__()
      self.conv = nn.Conv1d(1, 1, 1)
      self.relu = nn.ReLU()
      self.bn = nn.BatchNorm1d(1)
  
imports: from unittest import TestCase
  from mock import patch, Mock
  from torch import nn
  from torchbearer.callbacks.manifold_mixup import ManifoldMixup
  import torchbearer
  import torch
label_desc: Use image node_id 1 to create a new TestModule object from inherited base classes: nn with example: obj = TestModule()