code: def _build_dataset(file_path):
      path = epath.Path(file_path)
      with path.open("rb") as f:
          with h5py.File(f, "r") as dataset_file:
              data = dataset_file["data"]
              for key in data:
                  yield {
                      "steps": dataset_utils.build_episode(data[key])
                  }
  
file_path: datasets/tensorflow_datasets/robomimic/dataset_utils_test.py
imports: import inspect
  import os
  from absl import flags
  from absl.testing import absltest
  from etils import epath
  import h5py
  import numpy
  from tensorflow_datasets.robomimic import dataset_utils
  import tree
label_desc: Use image node_id 1 for calling a global function with example usage: _build_dataset(file_path) without return types