Rename parquet file format enum

tensorflow · May 27, 2022 · 3bdccaf · 3bdccaf
1 parent 30bec96
commit 3bdccaf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/tfx/components/util/tfxio_utils.py b/tfx/components/util/tfxio_utils.py
@@ -32,7 +32,10 @@
 from tensorflow_metadata.proto.v0 import schema_pb2
 
 
-_SUPPORTED_FILE_FORMATS = (example_gen_pb2.FileFormat.FORMAT_PARQUET, example_gen_pb2.FileFormat.FORMAT_TFRECORDS_GZIP)
+_SUPPORTED_FILE_FORMATS = (
+    example_gen_pb2.FileFormat.FILE_FORMAT_PARQUET,
+    example_gen_pb2.FileFormat.FORMAT_TFRECORDS_GZIP
+)
 # TODO(b/162532479): switch to support List[str] exclusively, once tfx-bsl
 # post-0.22 is released.
 OneOrMorePatterns = Union[str, List[str]]

diff --git a/tfx/proto/example_gen.proto b/tfx/proto/example_gen.proto
@@ -127,7 +127,7 @@ enum FileFormat {
 
   // Indicates parquet format files in any of the supported compressions.
   // https://arrow.apache.org/docs/python/parquet.html#compression-encoding-and-file-compatibility
-  FORMAT_PARQUET = 16;
+  FILE_FORMAT_PARQUET = 16;
 
   reserved 1 to 4, 7 to 15, 17 to max;
 }