code: def test_convert_module(precision, expected_dtype):
      precision = HalfPrecision(precision=precision)
      module = torch.nn.Linear(2, 2)
      assert module.weight.dtype == module.bias.dtype == torch.float32
      module = precision.convert_module(module)
      assert module.weight.dtype == module.bias.dtype == expected_dtype
  
imports: import pytest
  import torch
  from lightning.pytorch.plugins import HalfPrecision
label_desc: Use image node_id 4 for calling a global function with example usage: test_convert_module(precision, expected_dtype) without return types