Partial rollback of cl/315563442:

the flag TFDV_ACCEPT_RECORD_BATCH cannot be removed before tfx > 0.21.0 is released.

PiperOrigin-RevId: 318845937

Author: brills

RELEASE.md

@@ -5,6 +5,7 @@
 ## Major Features and Improvements
 
 ## Bug Fixes and Other Changes
+*   Fixed a bug that affected tfx 0.22.0 to work with TFDV 0.22.1.
 
 ## Known Issues
 

tensorflow_data_validation/__init__.py

@@ -84,3 +84,9 @@
 
 # Import version string.
 from tensorflow_data_validation.version import __version__
+
+# If tfdv module has this flag, tfdv.GenerateStatistics accepts
+# PCollection[pa.RecordBatch] as the input; otherwise it accepts
+# PCollection[pa.Table]
+# TODO(b/153368237): clean this up once tfx>0.22.0 is released.
+TFDV_ACCEPT_RECORD_BATCH = True