code: def __init__(self, num_outputs=2):
      super().__init__()
      self.linear1 = nn.Linear(1, 10)
      self.linear2 = nn.Linear(10, num_outputs)
      self.dist = Discrete()
  
imports: import pytest
  import wandb
label_desc: Use image node_id 1 to create a new DiscreteModel object from inherited base classes: nn with example: obj = DiscreteModel(num_outputs)