d8c2e2e64a92923f1448cf5302c92a80bb469a20,tensorflow_ranking/python/feature.py,,encode_listwise_features,#Any#Any#Any#Any#Any#Any#,109

Before Change

    reshaped_features = {}
    for name in example_feature_columns:
      if name not in features:
        continue
      batch_size = tf.shape(input=features[name])[0]
      try:
        reshaped_features[name] = utils.reshape_first_ndims(
            features[name], 2, [batch_size * input_size])

After Change

    reshaped_features = {}
    for name in example_specs:
      if name not in features:
        tf.compat.v1.logging.warn("Feature {} is not found.".format(name))
        continue
      batch_size = tf.shape(input=features[name])[0]
      try:
        reshaped_features[name] = utils.reshape_first_ndims(
            features[name], 2, [batch_size * input_size])

Instances