import tensorflow as tf
import tensorflow_data_validation as tfdv

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
/var/folders/x8/54gm3z5n28b76cvlkrl13j_80000gn/T/ipykernel_4855/3507982694.py in <module>
      1 import tensorflow as tf
----> 2 import tensorflow_data_validation as tfdv
      3 
      4 print('TF version: {}'.format(tf.__version__))
      5 print('TFDV version: {}'.format(tfdv.version.__version__))

~/opt/anaconda3/envs/tensorflow-2/lib/python3.8/site-packages/tensorflow_data_validation/__init__.py in <module>
     16 
     17 # Import stats API.
---> 18 from tensorflow_data_validation.api.stats_api import GenerateStatistics
     19 from tensorflow_data_validation.api.stats_api import WriteStatisticsToBinaryFile
     20 from tensorflow_data_validation.api.stats_api import WriteStatisticsToTFRecord

~/opt/anaconda3/envs/tensorflow-2/lib/python3.8/site-packages/tensorflow_data_validation/api/stats_api.py in <module>
     49 import apache_beam as beam
     50 import pyarrow as pa
---> 51 from tensorflow_data_validation.statistics import stats_impl
     52 from tensorflow_data_validation.statistics import stats_options
     53 

~/opt/anaconda3/envs/tensorflow-2/lib/python3.8/site-packages/tensorflow_data_validation/statistics/stats_impl.py in <module>
     25 from tensorflow_data_validation import types
     26 from tensorflow_data_validation.arrow import arrow_util
---> 27 from tensorflow_data_validation.statistics import stats_options
     28 from tensorflow_data_validation.statistics.generators import basic_stats_generator
     29 from tensorflow_data_validation.statistics.generators import image_stats_generator

~/opt/anaconda3/envs/tensorflow-2/lib/python3.8/site-packages/tensorflow_data_validation/statistics/stats_options.py in <module>
     30 from tensorflow_data_validation.utils import slicing_util
     31 from tfx_bsl.arrow import sql_util
---> 32 from tfx_bsl.coders import example_coder
     33 
     34 from google.protobuf import json_format

~/opt/anaconda3/envs/tensorflow-2/lib/python3.8/site-packages/tfx_bsl/coders/example_coder.py in <module>
     16 
     17 import pyarrow as pa
---> 18 from tfx_bsl.tfxio import tensor_representation_util
     19 
     20 from tensorflow_metadata.proto.v0 import schema_pb2

~/opt/anaconda3/envs/tensorflow-2/lib/python3.8/site-packages/tfx_bsl/tfxio/tensor_representation_util.py in <module>
     44 _LEGACY_DEFAULT_VALUE_FOR_FEATURE_TYPE = {
     45     schema_pb2.BYTES:
---> 46         schema_pb2.TensorRepresentation.DefaultValue(bytes_value=b""),
     47     schema_pb2.INT:
     48         schema_pb2.TensorRepresentation.DefaultValue(int_value=-1),

AttributeError: module 'tensorflow_metadata.proto.v0.schema_pb2' has no attribute 'TensorRepresentation'