code: def get_graph_build_func(build_method):
      if build_method.lower() == "zero_filling":
          return graph_build_zero_filling
      elif build_method.lower() == "split":
          return graph_build_split
      else:
          raise NotImplementedError
  
imports: import copy
  import torch
  import numpy
  from scipy.special import comb
  from itertools import combinations
  import torch.nn.functional
  from torch_geometric.utils import to_networkx
  from torch_geometric.data import Data, Batch, Dataset
  from torch_geometric.loader import DataLoader
label_desc: Use image node_id 3 for calling a global function with example usage: get_graph_build_func(build_method) and returns: graph_build_zero_filling, graph_build_split