code: def test_infection_dataset_reproducibility():
      graph_generator = ERGraph(num_nodes=500, edge_prob=0.004)
  
      seed_everything(12345)
      dataset1 = InfectionDataset(
          graph_generator, num_infected_nodes=50, max_path_length=5
      )
  
      seed_everything(12345)
      dataset2 = InfectionDataset(
          graph_generator, num_infected_nodes=50, max_path_length=5
      )
  
      assert torch.equal(dataset1[0].edge_mask, dataset2[0].edge_mask)
  
file_path: pytorch_geometric/test/datasets/test_infection_dataset.py
imports: import torch
  from torch_geometric import seed_everything
  from torch_geometric.data import Data
  from torch_geometric.datasets import InfectionDataset
  from torch_geometric.datasets.graph_generator import ERGraph, GraphGenerator
label_desc: Use image node_id 2 for calling a global function with example usage: test_infection_dataset_reproducibility() without return types