Fix pytype for stats_gen_lib.

PiperOrigin-RevId: 235076509

Author: tf-data-validation-team

tensorflow_data_validation/utils/stats_gen_lib.py

@@ -34,7 +34,7 @@
 from tensorflow_data_validation.coders import tf_example_decoder
 from tensorflow_data_validation.statistics import stats_impl
 from tensorflow_data_validation.statistics import stats_options as options
-from tensorflow_data_validation.types_compat import List, Optional
+from tensorflow_data_validation.types_compat import List, Optional, Text
 
 from tensorflow_metadata.proto.v0 import schema_pb2
 from tensorflow_metadata.proto.v0 import statistics_pb2