code: def postprocess(batch, task):
      """Post process the batch and make it ready to be sent to the model."""
      if task == "stsb":  # Regression task
          return dict(targets=batch[1][:, None], **batch[0])
      return dict(label=batch[1], **batch[0])
  
file_path: scenic/scenic/projects/baselines/bert/datasets/bert_glue_dataset.py
imports: import functools
  import json
  from typing import Optional
  from absl import logging
  from flax import jax_utils
  import jax
  import jax.numpy
  from scenic.dataset_lib import dataset_utils
  from scenic.dataset_lib import datasets
  import tensorflow
  from official.nlp.data.bert import input_pipeline
label_desc: Use image node_id 3 for calling a global function with example usage: postprocess(batch, task) and returns: dict, dict