08c0e994cc0e20cb576469c01b472a43e3279e86,tensorflow_transform/beam/tft_beam_io/transform_fn_io_test.py,,,#,39

Before Change

from tensorflow.python.lib.io import file_io

_TEST_METADATA_COMPLETE = dataset_metadata.DatasetMetadata({
    "fixed_column": dataset_schema.ColumnSchema(
        tf.string, (3,), dataset_schema.FixedColumnRepresentation()),
    "list_columm": dataset_schema.ColumnSchema(
        tf.float32, (None,), dataset_schema.ListColumnRepresentation())
})

After Change

    self.assertTrue(file_io.file_exists(transform_fn_dir))
    self.assertTrue(file_io.is_directory(transform_fn_dir))

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

Instances