code: def __init__(self):
      self.block1 = ModelBlockWithArg()
      self.block2 = ModelBlockWithArg()
      self.op1 = OpWithArg()
  
imports: import functools
  import inspect
  import unittest
  from typing import Tuple
  import numpy
  from jax import numpy
  import objax
  from objax.typing import JaxArray
label_desc: Use image node_id 1 to create a new ModelWithArg object from inherited base classes: objax with example: obj = ModelWithArg()