code: def _context_feature_columns():
      return {
          "query_length": tf.feature_column.numeric_column(
              "query_length",
              shape=(1,),
              default_value=0,
              dtype=tf.int64,
          )
      }
  
imports: from absl.testing import parameterized
  import tensorflow
  from google.protobuf import text_format
  from tensorflow_ranking.python import data
  from tensorflow_ranking.python.keras import feature
  from tensorflow_ranking.python.keras import losses
  from tensorflow_ranking.python.keras import metrics
  from tensorflow_ranking.python.keras import model
  from tensorflow_ranking.python.keras import network
  from tensorflow_serving.apis import input_pb2
label_desc: Use image node_id 1 for calling a global function with example usage: _context_feature_columns() and returns: dict