08c0e994cc0e20cb576469c01b472a43e3279e86,tensorflow_transform/beam/tft_beam_io/beam_metadata_io_test.py,,,#,33

Before Change

_TEST_METADATA_COMPLETE = dataset_metadata.DatasetMetadata({
    "fixed_column": dataset_schema.ColumnSchema(
        tf.string, (3,), dataset_schema.FixedColumnRepresentation()),
    "list_columm": dataset_schema.ColumnSchema(
        dataset_schema.IntDomain(tf.int64, min_value=-1, max_value=5),
        (None,), dataset_schema.ListColumnRepresentation())
})

_TEST_METADATA = dataset_metadata.DatasetMetadata({
    "fixed_column": dataset_schema.ColumnSchema(

After Change

    self.assertEqual(metadata, test_metadata.COMPLETE_METADATA)


if __name__ == "__main__":
  unittest.main()

Instances