Add video support for CroissantBuilder.

PiperOrigin-RevId: 788928190

Author: The TensorFlow Datasets Authors

tensorflow_datasets/core/dataset_builders/croissant_builder.py

@@ -58,6 +58,7 @@
 from tensorflow_datasets.core.features import sequence_feature
 from tensorflow_datasets.core.features import tensor_feature
 from tensorflow_datasets.core.features import text_feature
+from tensorflow_datasets.core.features import video_feature
 from tensorflow_datasets.core.utils import conversion_utils
 from tensorflow_datasets.core.utils import croissant_utils
 from tensorflow_datasets.core.utils import type_utils
@@ -195,6 +196,8 @@ def datatype_converter(
     feature = audio_feature.Audio(
         doc=field.description, sample_rate=field.source.sampling_rate
     )
+  elif field_data_type == mlc.DataType.VIDEO_OBJECT:
+    feature = video_feature.Video(doc=field.description)
   else:
     raise ValueError(
         f'Unknown data type: {field_data_type} for field {field.id}.'

tensorflow_datasets/core/dataset_builders/croissant_builder_test.py

@@ -28,6 +28,7 @@
 from tensorflow_datasets.core.features import sequence_feature
 from tensorflow_datasets.core.features import tensor_feature
 from tensorflow_datasets.core.features import text_feature
+from tensorflow_datasets.core.features import video_feature
 from tensorflow_datasets.core.utils.lazy_imports_utils import mlcroissant as mlc
 
 FileFormat = file_adapters.FileFormat
@@ -225,6 +226,14 @@ def test_datatype_converter_bbox_with_invalid_format():
             bounding_boxes.BBoxFeature,
             None,
         ),
+        (
+            mlc.Field(
+                data_types=mlc.DataType.VIDEO_OBJECT,
+                description="Video feature",
+            ),
+            video_feature.Video,
+            None,
+        ),
         (
             mlc.Field(
                 id="person",