code: def _add_ragged_label(inputs):
      mask = inputs[_MASK]
      features_dict = {
          "query_tokens": inputs["query_tokens"],
          "document_tokens": inputs["document_tokens"],
      }
      label = tf.ragged.boolean_mask(inputs[_LABEL_FEATURE], mask)
      label = tf.cast(label, dtype=tf.float32)
      return features_dict, label
  
imports: import os
  from typing import List
  from absl import app
  from absl import flags
  from absl import logging
  import tensorflow
  import tensorflow_ranking
label_desc: Use image node_id 1 for calling a global function with example usage: _add_ragged_label(inputs) and returns: features_dict,  label