class_docstr: Robot pushing dataset from BAIR (Small 64x64 version).
code: def _info(self):
      # The Bair dataset consist of a sequence of frames (video) with associated
      # metadata (action and position)
      features = tfds.features.Sequence(
          {
              "image_main": tfds.features.Image(shape=IMG_SHAPE),
              "image_aux1": tfds.features.Image(shape=IMG_SHAPE),
              "action": tfds.features.Tensor(
                  shape=(4,), dtype=np.float32
              ),
              "endeffector_pos": tfds.features.Tensor(
                  shape=(3,), dtype=np.float32
              ),
          },
          length=FRAMES_PER_VIDEO,
      )
  
      return self.dataset_info_from_configs(
          features=features,
          homepage="https://sites.google.com/view/sna-visual-mpc/",
      )
  
file_path: datasets/tensorflow_datasets/datasets/bair_robot_pushing_small/bair_robot_pushing_small_dataset_builder.py
imports: import os
  from absl import logging
  import numpy
  from tensorflow_datasets.core.utils.lazy_imports_utils import tensorflow
  import tensorflow_datasets.public_api
label_desc: Use image node_id 1 for calling the Builder obj's underlying member method code with example usage: obj._info() and returns: self