code: def test_from_state_dict():
      model = torch.nn.Linear(1, 1)
      expected_state_dict = model.state_dict()
      checkpoint = TorchCheckpoint.from_state_dict(expected_state_dict)
      actual_state_dict = checkpoint.get_model(
          torch.nn.Linear(1, 1)
      ).state_dict()
      assert actual_state_dict == expected_state_dict
  
imports: import torch
  from ray.train.torch import TorchCheckpoint
label_desc: Use image node_id 3 for calling a global function with example usage: test_from_state_dict() without return types