code: def test_pemsbay_task_generator():
      loader = PemsBayDatasetLoader(raw_data_dir="/tmp/")
      dataset = loader.get_dataset(
          num_timesteps_in=6, num_timesteps_out=5
      )
      for epoch in range(2):
          for snapshot in dataset:
              assert snapshot.edge_index.shape == (2, 2694)
              assert snapshot.edge_attr.shape == (2694,)
              assert snapshot.x.shape == (325, 2, 6)
              assert snapshot.y.shape == (325, 2, 5)
  
imports: import numpy
  import networkx
  from torch_geometric_temporal.signal import temporal_signal_split
  from torch_geometric_temporal.signal import StaticGraphTemporalSignal
  from torch_geometric_temporal.signal import DynamicGraphTemporalSignal
  from torch_geometric_temporal.signal import DynamicGraphStaticSignal
  from torch_geometric_temporal.signal import StaticHeteroGraphTemporalSignal
  from torch_geometric_temporal.signal import DynamicHeteroGraphTemporalSignal
  from torch_geometric_temporal.signal import DynamicHeteroGraphStaticSignal
  from torch_geometric_temporal.dataset import METRLADatasetLoader, PemsBayDatasetLoader
  from torch_geometric_temporal.dataset import ChickenpoxDatasetLoader, PedalMeDatasetLoader, WikiMathsDatasetLoader, EnglandCovidDatasetLoader
  from torch_geometric_temporal.dataset import TwitterTennisDatasetLoader, MontevideoBusDatasetLoader, MTMDatasetLoader
  from torch_geometric_temporal.dataset import WindmillOutputLargeDatasetLoader, WindmillOutputMediumDatasetLoader, WindmillOutputSmallDatasetLoader
label_desc: Use image node_id 35 for calling a global function with example usage: test_pemsbay_task_generator() without return types