code: def set_dataset_info(datasets):
      r"""
      Set global dataset information
  
      Args:
          datasets: List of dataset object
  
      """
      # get dim_in and dim_out
      try:
          cfg.share.dim_in = datasets[0].num_node_features
      except Exception:
          cfg.share.dim_in = 1
      try:
          cfg.share.dim_out = datasets[0].num_labels
          if (
              "classification" in cfg.dataset.task_type
              and cfg.share.dim_out == 2
          ):
              cfg.share.dim_out = 1
      except Exception:
          cfg.share.dim_out = 1
  
      # count number of dataset splits
      cfg.share.num_splits = len(datasets)
  
imports: import logging
  import pickle
  import time
  import networkx
  import torch
  import torch_geometric.transforms
  from deepsnap.batch import Batch
  from deepsnap.dataset import GraphDataset
  from ogb.graphproppred import PygGraphPropPredDataset
  from torch.utils.data import DataLoader
  from torch_geometric.datasets import PPI, Amazon, Coauthor, KarateClub, MNISTSuperpixels, Planetoid, QM7b, TUDataset
  import graphgym.models.feature_augment
  import graphgym.register
  from graphgym.config import cfg
  from graphgym.models.transform import edge_nets, ego_nets, path_len, remove_node_feature
label_desc: Use image node_id 7 for calling a global function with example usage: set_dataset_info(datasets) without return types