code: def _dataset_info(
      data_dir: str, name: str = "test_dataset"
  ) -> dataset_info.DatasetInfo:
      return dataset_info.DatasetInfo(
          builder=dataset_info.DatasetIdentity(
              name=name,
              data_dir=data_dir,
              module_name="",
              version="1.0.0",
          ),
          description="Test builder",
          features=tfds.features.FeaturesDict(
              {
                  "a": tf.int32,
                  "b": tf.int32,
              }
          ),
      )
  
imports: import tempfile
  from absl.testing import absltest
  from absl.testing import parameterized
  import tensorflow
  import tensorflow_datasets
  from tensorflow_datasets.core import dataset_info
  from tensorflow_datasets.core import sequential_writer
label_desc: Use image node_id 2 for calling a global function with example usage: _dataset_info(data_dir, name) and returns: dataset_info