code: def test_parametrized_module_inheritance():
      class Module1(ParametrizedModule):
          def __init__(self, x):
              super().__init__()
              self.t = x
  
      class Module2(Module1):
          def __init__(self, y):
              super().__init__(y)
  
      choice = nni.choice("choice1", [1, 2, 3])
      with frozen_context():
          module = Module2(choice)
      assert _mutable_equal(module.args, {"y": choice})
      assert module.freeze({"choice1": 2}).t == 2
  
imports: import copy
  import pytest
  import torch
  import nni
  from torch import nn
  from nni.common.serializer import is_traceable
  from nni.mutable import Categorical, ExpressionConstraint, ensure_frozen, frozen_context, label_scope, auto_label, SampleMissingError
  from nni.mutable.mutable import _mutable_equal
  from nni.nas.nn.pytorch import ModelSpace, MutableModule, ParametrizedModule
  from nni.nas.space import model_context
label_desc: Use image node_id 21 for calling a global function with example usage: test_parametrized_module_inheritance() without return types