code: def test_lazy_load_module(tmp_path):
      model0 = nn.Linear(2, 2)
      torch.save(model0.state_dict(), tmp_path / "model.pt")
  
      model1 = nn.Linear(2, 2)
      checkpoint = _lazy_load(tmp_path / "model.pt")
      model1.load_state_dict(checkpoint)
  
      assert isinstance(checkpoint["weight"], _NotYetLoadedTensor)
      assert type(model0.weight.data) is torch.Tensor
      assert torch.equal(model0.weight, model1.weight)
      assert torch.equal(model0.bias, model1.bias)
  
file_path: pytorch-lightning/tests/tests_fabric/utilities/test_load.py
imports: import pytest
  import torch
  import torch.nn
  from lightning.fabric.utilities.load import _lazy_load, _materialize_tensors, _move_state_into, _NotYetLoadedTensor
  from tests_fabric.helpers.runif import RunIf
label_desc: Use image node_id 1 for calling a global function with example usage: test_lazy_load_module(tmp_path) without return types