code: def set_weight(self, weight: torch.Tensor) -> None:
      self.weight[:] = torch.nn.Parameter(weight)
  
file_path: torchrec/torchrec/inference/tests/test_modules.py
imports: import torch
label_desc: Use image node_id 3 for calling the Simple obj's underlying member method code with example usage: obj.set_weight(weight) without return types