code: def __init__(self, bias) -> None:
      super().__init__()
      self.linear_1 = LinearModel(3, 3, bias)
      self.linear_2 = LinearModel(3, 3, bias)
      self.conv = ConvModel(3, 6, 3, bias)
  
file_path: ColossalAI/tests/test_analyzer/test_fx/test_mod_dir.py
imports: import pytest
  import torch
  from colossalai.testing import clear_cache_before_run, parameterize
label_desc: Use image node_id 1 to create a new AModel object from inherited base classes: torch.nn with example: obj = AModel(bias)