class_docstr: Task object for tf-ranking BERT.
class_docstr_tok: ["Task","object","for","tf-ranking","BERT","."]
docstr: Returns tf.data.Dataset for tf-ranking BERT task.
code: def build_inputs(self, params, input_context=None):
      """Returns tf.data.Dataset for tf-ranking BERT task."""
      return TFRBertDataLoader(
          params,
          label_spec=self._label_spec,
          dataset_fn=self._dataset_fn,
      ).load(input_context)
  
imports: import dataclasses
  import os
  import time
  from typing import Dict, Mapping, Optional, Tuple
  from absl import logging
  import numpy
  import tensorflow
  from official.core import task_factory
  from official.modeling.hyperparams import base_config
  from official.nlp.configs import encoders
  from tensorflow_ranking.extension import task
  from tensorflow_ranking.python.keras import metrics
  from tensorflow_ranking.python.keras import model
label_desc: Use image node_id 3 for calling the TFRBertTask obj's underlying member method code with example usage: obj.build_inputs(params, input_context) and returns: TFRBertDataLoader