code: def load_data(dataset):
      return _DATASETS[dataset]
  
imports: import argparse
  import textwrap
  import tensorflow
  import tensorflow_addons
  from dgl.data import CiteseerGraphDataset, CoraGraphDataset, PubmedGraphDataset
  from dgl.nn.tensorflow.conv import SGConv
label_desc: Use image node_id 1 for calling a global function with example usage: load_data(dataset) and returns: _DATASETS