code: def get_transform(name):
      transform_funcs = {
          "ego": ego_nets,
          "path": path_len,
      }
      assert (
          name in transform_funcs.keys()
      ), "Transform function '{}' not supported".format(name)
      return transform_funcs[name]
  
imports: import torch
  import argparse
  import skip_models
  import numpy
  import torch.nn
  import torch.nn.functional
  import torch_geometric.transforms
  import torch_geometric.nn
  import torch.optim
  from torch.utils.data import DataLoader
  from torch_geometric.datasets import TUDataset
  from sklearn.metrics import *
  from torch.nn import Sequential, Linear, ReLU
  from deepsnap.dataset import GraphDataset
  from deepsnap.batch import Batch
  from transforms import *
label_desc: Use image node_id 2 for calling a global function with example usage: get_transform(name) and returns: transform_funcs