class_docstr: DatasetBuilder for robomimic_mh dataset.
code: def _get_metadata(self) -> Dict[Any, Any]:
      if self.builder_config.task == utils.Task.TRANSPORT:
          episode_metadata = {
              "20_percent": np.bool_,
              "20_percent_train": np.bool_,
              "20_percent_valid": np.bool_,
              "50_percent": np.bool_,
              "50_percent_train": np.bool_,
              "50_percent_valid": np.bool_,
              "better": np.bool_,
              "better_train": np.bool_,
              "better_valid": np.bool_,
              "okay": np.bool_,
              "okay_better": np.bool_,
              "okay_better_train": np.bool_,
              "okay_better_valid": np.bool_,
              "okay_train": np.bool_,
              "okay_valid": np.bool_,
              "train": np.bool_,
              "valid": np.bool_,
              "worse": np.bool_,
              "worse_better": np.bool_,
              "worse_better_train": np.bool_,
              "worse_better_valid": np.bool_,
              "worse_okay": np.bool_,
              "worse_okay_train": np.bool_,
              "worse_okay_valid": np.bool_,
              "worse_train": np.bool_,
              "worse_valid": np.bool_,
          }
      else:
          episode_metadata = {
              "20_percent": np.bool_,
              "20_percent_train": np.bool_,
              "20_percent_valid": np.bool_,
              "50_percent": np.bool_,
              "50_percent_train": np.bool_,
              "50_percent_valid": np.bool_,
              "better": np.bool_,
              "better_operator_1": np.bool_,
              "better_operator_1_train": np.bool_,
              "better_operator_1_valid": np.bool_,
              "better_operator_2": np.bool_,
              "better_operator_2_train": np.bool_,
              "better_operator_2_valid": np.bool_,
              "better_train": np.bool_,
              "better_valid": np.bool_,
              "okay": np.bool_,
              "okay_better": np.bool_,
              "okay_better_train": np.bool_,
              "okay_better_valid": np.bool_,
              "okay_operator_1": np.bool_,
              "okay_operator_1_train": np.bool_,
              "okay_operator_1_valid": np.bool_,
              "okay_operator_2": np.bool_,
              "okay_operator_2_train": np.bool_,
              "okay_operator_2_valid": np.bool_,
              "okay_train": np.bool_,
              "okay_valid": np.bool_,
              "train": np.bool_,
              "valid": np.bool_,
              "worse": np.bool_,
              "worse_better": np.bool_,
              "worse_better_train": np.bool_,
              "worse_better_valid": np.bool_,
              "worse_okay": np.bool_,
              "worse_okay_train": np.bool_,
              "worse_okay_valid": np.bool_,
              "worse_operator_1": np.bool_,
              "worse_operator_1_train": np.bool_,
              "worse_operator_1_valid": np.bool_,
              "worse_operator_2": np.bool_,
              "worse_operator_2_train": np.bool_,
              "worse_operator_2_valid": np.bool_,
              "worse_train": np.bool_,
              "worse_valid": np.bool_,
          }
      return episode_metadata
  
imports: from __future__ import annotations
  from typing import Any, Dict
  import numpy
  import tensorflow_datasets.public_api
  from tensorflow_datasets.robomimic import dataset_utils
label_desc: Use image node_id 1 for calling the Builder obj's underlying member method code with example usage: obj._get_metadata() and returns: episode_metadata