code: def test_predict_module(self) -> None:
      module = TestModule()
      predict_module = TestPredictModule(module)
  
      module_state_dict = module.state_dict()
      predict_module_state_dict = predict_module.state_dict()
  
      self.assertEqual(
          module_state_dict.keys(), predict_module_state_dict.keys()
      )
  
      for tensor0, tensor1 in zip(
          module_state_dict.values(), predict_module_state_dict.values()
      ):
          self.assertTrue(torch.equal(tensor0, tensor1))
  
file_path: torchrec/torchrec/inference/tests/predict_module_tests.py
imports: import unittest
  from typing import Dict
  import torch
  import torch.nn
  from torchrec.inference.modules import PredictModule, quantize_dense
label_desc: Use image node_id 1 for calling the PredictModulesTest obj's underlying member method code with example usage: obj.test_predict_module() without return types