code: def __init__(self, metadata):
      super().__init__()
      self.graph_sage = to_hetero(GraphSAGE(), metadata, debug=False)
      self.lin = torch.nn.Linear(32, 1)
  
file_path: pytorch_geometric/test/explain/conftest.py
imports: from typing import Optional
  import pytest
  import torch
  from torch_geometric.data import Data, HeteroData
  from torch_geometric.explain import Explanation
  from torch_geometric.explain.config import MaskType
  from torch_geometric.nn import SAGEConv, to_hetero
  from torch_geometric.testing import get_random_edge_index
label_desc: Use image node_id 1 to create a new HeteroSAGE object from inherited base classes: torch.nn with example: obj = HeteroSAGE(metadata)