bc0c3b6035aea0b25b827e46d4a67a1e3bc9078e,tensorflow_transform/saved/input_fn_maker_test.py,,_make_raw_schema,#Any#Any#,35

Before Change

  schema.column_schemas["raw_b"] = (sch.ColumnSchema(
      tf.int64, shape, sch.FixedColumnRepresentation(default_value=1)))

  schema.column_schemas["raw_label"] = (sch.ColumnSchema(
      tf.int64, shape, sch.FixedColumnRepresentation(default_value=-1)))

  if should_add_unused_feature:
    schema.column_schemas["raw_unused"] = (sch.ColumnSchema(
        tf.int64, shape, sch.FixedColumnRepresentation(default_value=1)))

After Change

def _make_raw_schema(
    shape=None,
    should_add_unused_feature=False):
  feature_spec = {
      "raw_a": tf.FixedLenFeature(shape, tf.int64, 0),
      "raw_b": tf.FixedLenFeature(shape, tf.int64, 1),
      "raw_label": tf.FixedLenFeature(shape, tf.int64, -1),
  }
  if should_add_unused_feature:
    feature_spec["raw_unused"] = tf.FixedLenFeature(shape, tf.int64, 1)
  return _MockSchema(feature_spec=feature_spec)


def _make_transformed_schema(shape):
  feature_spec = {

Instances