code: def fake_hetero_data(node_types=2, edge_types=5) -> HeteroData:
      return FakeHeteroDataset(
          num_node_types=node_types,
          num_edge_types=edge_types,
          avg_num_nodes=10,
          edge_dim=2,
      )[0]
  
imports: import numbers
  from typing import Dict, Generator, List, Optional, Tuple, Union
  import pytest
  import torch
  from torch_geometric.data import Data, HeteroData
  from torch_geometric.datasets import FakeDataset, FakeHeteroDataset
  from torch_geometric.transforms import Pad
  from torch_geometric.transforms.pad import AttrNamePadding, EdgeTypePadding, NodeTypePadding, Padding, UniformPadding
  from torch_geometric.typing import EdgeType, NodeType
label_desc: Use image node_id 2 for calling a global function with example usage: fake_hetero_data(node_types, edge_types) and returns: FakeHeteroDataset