code: def __init__(self, foo, bar):
      super().__init__()
      self.foo = nn.Linear(foo, 3)
      self.bar = nn.Dropout(bar)
  
imports: import nni.retiarii.nn.pytorch
  from nni.retiarii import basic_unit
label_desc: Use image node_id 1 to create a new ImportTest object from inherited base classes: nn with example: obj = ImportTest(foo, bar)