bc0c3b6035aea0b25b827e46d4a67a1e3bc9078e,tensorflow_transform/saved/input_fn_maker_test.py,,_make_transformed_schema,#Any#,54

Before Change

def _make_transformed_schema(shape):
  schema = sch.Schema()

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

  schema.column_schemas["transformed_b"] = (
      sch.ColumnSchema(tf.int64, shape, sch.ListColumnRepresentation()))

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

  return schema


class InputFnMakerTest(unittest.TestCase):

After Change

def _make_transformed_schema(shape):
  feature_spec = {
      "transformed_a": tf.FixedLenFeature(shape, tf.int64),
      "transformed_b": tf.VarLenFeature(tf.int64),
      "transformed_label": tf.FixedLenFeature(shape, tf.int64),
  }
  return _MockSchema(feature_spec=feature_spec)


class InputFnMakerTest(unittest.TestCase):

Instances