192a807b39bc029c273c2d1349fb73850ed9484c,nni/retiarii/nn/pytorch/nn.py,ModuleList,__init__,#ModuleList#,99

Before Change

class ModuleList(nn.ModuleList):
    def __init__(self, *args):
        global _records
        if _records is not None:
            _records[id(self)] = {} // no args need to be recorded
        super(ModuleList, self).__init__(*args)

def wrap_module(original_class):
    orig_init = original_class.__init__

After Change

class ModuleList(nn.ModuleList):
    def __init__(self, *args):
        add_record(id(self), {})
        super(ModuleList, self).__init__(*args)

def wrap_module(original_class):
    orig_init = original_class.__init__

Instances