Fix documentation for the Video feature.

PiperOrigin-RevId: 788915522

Author: The TensorFlow Datasets Authors

tensorflow_datasets/core/features/video_feature.py

@@ -129,14 +129,15 @@ def __init__(
       frame_shape = shape[1:]
     self._encoding_format = encoding_format
     self._extra_ffmpeg_args = list(ffmpeg_extra_args or [])
-    super(Video, self).__init__(
+    super().__init__(
         image_feature.Image(
             shape=frame_shape,
             dtype=dtype,
             encoding_format=encoding_format,
             use_colormap=use_colormap,
         ),
         length=shape[0] if shape else None,
+        doc=doc,
     )
 
   def _ffmpeg_decode(self, path_or_fobj):

tensorflow_datasets/core/features/video_feature_test.py

@@ -26,6 +26,7 @@
 from tensorflow_datasets.core import features
 
 
+
 class VideoFeatureTest(testing.FeatureExpectationsTestCase):
 
   @property
@@ -34,9 +35,10 @@ def _test_data_path(self):
 
   def test_video_numpy(self):
     np_video = np.random.randint(256, size=(128, 64, 64, 3), dtype=np.uint8)
-
+    doc = 'This is a test video.'
+    feature = features.Video(shape=(None, 64, 64, 3), doc=doc)
     self.assertFeature(
-        feature=features.Video(shape=(None, 64, 64, 3)),
+        feature=feature,
         shape=(None, 64, 64, 3),
         dtype=tf.uint8,
         tests=[
@@ -135,5 +137,6 @@ def read(self, *args, **kwargs):
           ],
       )
 
+
 if __name__ == '__main__':
   testing.test_main()