code: def get_dataset(data_name, split, args):
      return dataset_dict[data_name](split, args)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from os import path
  import tensorflow.compat.v1
label_desc: Use image node_id 1 for calling a global function with example usage: get_dataset(data_name, split, args) and returns: dataset_dict