code: def test_hetero_dict_linear_jit():
      x_dict = {
          "v": torch.randn(3, 16),
          "w": torch.randn(2, 8),
      }
  
      lin = HeteroDictLinear({"v": 16, "w": 8}, 32)
  
      jit = torch.jit.script(lin)
      assert len(jit(x_dict)) == 2
  
imports: import copy
  import warnings
  from typing import List
  import pytest
  import torch
  from torch import Tensor
  from torch.nn import Linear
  from torch.nn.parameter import UninitializedParameter
  import torch_geometric.backend
  from torch_geometric.nn import HeteroDictLinear, HeteroLinear, Linear
  from torch_geometric.profile import benchmark
  from torch_geometric.testing import withCUDA, withPackage
  from torch_geometric.typing import pyg_lib
  from torch_geometric.utils import cumsum
label_desc: Use image node_id 12 for calling a global function with example usage: test_hetero_dict_linear_jit() without return types