class_docstr: test for:
      prim::If
class_docstr_tok: ["test","for",":","prim",":",":If"]
code: def __init__(self):
      super().__init__()
      self.the_cond = True
  
imports: import unittest
  import torch
  import torch.nn.functional
  from nni.compression.pruning import L1NormPruner
  from nni.compression.speedup import ModelSpeedup
  from nni.compression.utils import compute_sparsity_compact2origin, compute_sparsity_mask2compact
label_desc: Use image node_id 1 to create a new CondModel object from inherited base classes: torch.nn with example: obj = CondModel()