code: def test_source_code_parity_on_infer_parameters(self) -> None:
      original_infer_parameters_src = inspect.getsource(
          LazyModuleMixin._infer_parameters
      )
      lazy_ext_infer_parameters_src = inspect.getsource(
          LazyModuleExtensionMixin._infer_parameters
      )
  
      # remove comments
      original_infer_parameters_src = remove_comment(
          original_infer_parameters_src
      )
      lazy_ext_infer_parameters_src = remove_comment(
          lazy_ext_infer_parameters_src
      )
  
      # reproduce the only changes:
      expected_lazy_ext_infer_parameters_src = original_infer_parameters_src.replace(
          "def _infer_parameters(self: _LazyProtocol, module, args, kwargs=None):",
          "def _infer_parameters(self: _LazyExtensionProtocol, module, args, kwargs) -> None:",
      ).replace(
          "module.initialize_parameters(*args)",
          "module.initialize_parameters(*args, **kwargs)",
      )
      self.assertEqual(
          lazy_ext_infer_parameters_src,
          expected_lazy_ext_infer_parameters_src,
          "Please make sure `LazyModuleExtensionMixin._infer_parameters` has the same source "
          "code as `LazyModuleMixin._infer_parameters` except the expected difference that "
          "is checked in this unit test.",
      )
  
file_path: torchrec/torchrec/modules/tests/test_lazy_extension.py
imports: import inspect
  import re
  import unittest
  from typing import Tuple
  import torch
  from torch.nn.modules.lazy import LazyModuleMixin
  from torchrec.modules.lazy_extension import lazy_apply, LazyModuleExtensionMixin
label_desc: Use image node_id 2 for calling the TestLazyModuleExtensionMixin obj's underlying member method code with example usage: obj.test_source_code_parity_on_infer_parameters() without return types