code: def setUp(self):
      super(KerasModelToEstimatorTest, self).setUp()
      (
          context_feature_columns,
          example_feature_columns,
          custom_objects,
      ) = _get_feature_columns()
      self._context_feature_columns = context_feature_columns
      self._example_feature_columns = example_feature_columns
      # Remove label feature from example feature column.
      del self._example_feature_columns[_LABEL_FEATURE]
  
      self._custom_objects = custom_objects
      self._network = _DummyUnivariateRankingNetwork(
          context_feature_columns=self._context_feature_columns,
          example_feature_columns=self._example_feature_columns,
      )
      self._loss = losses.get(
          losses.RankingLossKey.SOFTMAX_LOSS,
          reduction=tf.compat.v2.losses.Reduction.SUM_OVER_BATCH_SIZE,
      )
      self._eval_metrics = metrics.default_keras_metrics()
      self._optimizer = tf.keras.optimizers.legacy.Adagrad(
          learning_rate=0.1
      )
      self._config = tf_estimator.RunConfig(
          keep_checkpoint_max=2, save_checkpoints_secs=2
      )
  
      self._data_file = os.path.join(
          tf.compat.v1.test.get_temp_dir(), "test_elwc.tfrecord"
      )
      serialized_elwc_list = [
          _ELWC_PROTO.SerializeToString(),
      ] * 20
      if tf.io.gfile.exists(self._data_file):
          tf.io.gfile.remove(self._data_file)
      with tf.io.TFRecordWriter(self._data_file) as writer:
          for serialized_elwc in serialized_elwc_list:
              writer.write(serialized_elwc)
  
imports: import os
  from absl.testing import parameterized
  import tensorflow
  from tensorflow import estimator
  from tensorflow.compat.v1 import estimator
  from google.protobuf import text_format
  from tensorflow_ranking.python import data
  from tensorflow_ranking.python.keras import estimator
  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 the KerasModelToEstimatorTest obj's underlying member method code with example usage: obj.setUp() without return types