code: def get_glue_task_config(task_name, batch_size):
      """Returns GLUE task config."""
      config = ml_collections.ConfigDict()
      config.dataset_name = "bert_glue"
      config.data_dtype_str = "float32"
      config.batch_size = batch_size
      config.dataset_configs = ml_collections.ConfigDict()
      config.dataset_configs.task = task_name
      task_path = glue_common.GLUE_TASK_PATH[task_name]
      config.dataset_configs.input_meta_data_path = (
          glue_common.INPUT_MEAT_DATA_PATH.format(task_path=task_path)
      )
      config.dataset_configs.train_data_path = (
          glue_common.TRAIN_DATA_PATH.format(task_path=task_path)
      )
      config.dataset_configs.eval_data_path = (
          glue_common.EVAL_DATA_PATH.format(task_path=task_path)
      )
      config.dataset_configs.prefetch_to_device = 2
      return config
  
label_desc: Use image node_id 1 for calling a global function with example usage: get_glue_task_config(task_name, batch_size) and returns: config