import tensorflow as tf

          message = annotations_pb2.BucketBoundaries()
          annotation.Unpack(message)
          self.assertAllClose(list(message.boundaries), [1])

  def test_infer_feature_schema_with_ragged_tensor(self):
    with tf.compat.v1.Graph().as_default() as graph:
      outputs = {
          'foo': tf.RaggedTensor.from_row_splits(
              values=tf.constant([3, 1, 4, 1, 5, 9, 2, 6], tf.int64),
              row_splits=[0, 4, 4, 7, 8, 8]),
