def test_sequence_feature_column_name_not_struct_in_schema(self):
schema_text_proto = """
feature {
name: "##SEQUENCE##"
type: INT
}
"""
serialized_schema = text_format.Parse(
schema_text_proto, schema_pb2.Schema()).SerializeToString()
error_msg_regex = (
"Found a feature in the schema with the sequence_feature_column_name "
r"\(i.e., ##SEQUENCE##\) that is not a struct.*")
with self.assertRaisesRegex(RuntimeError, error_msg_regex):
sequence_example_coder.SequenceExamplesToRecordBatchDecoder(
_TEST_SEQUENCE_COLUMN_NAME, serialized_schema)
def _ArrowSchemaNoRawRecordColumn(self) -> pa.Schema:
return sequence_example_coder.SequenceExamplesToRecordBatchDecoder(
_SEQUENCE_COLUMN_NAME,
self._schema.SerializeToString(),
use_large_types=self._can_produce_large_types).ArrowSchema()