class_docstr: DatasetBuilder for rlu_control_suite dataset.
class_docstr_tok: ["DatasetBuilder","for","rlu_control_suite","dataset","."]
code: def get_features_dict(self):
      return tfds.features.FeaturesDict(
          {
              "steps": tfds.features.Dataset(
                  {
                      "observation": {
                          k: rlu_common.float_tensor_feature(v)
                          for k, v in self.builder_config.observation_size.items()
                      },
                      "action": tfds.features.Tensor(
                          shape=(self.builder_config.action_size,),
                          dtype=np.float32,
                      ),
                      "reward": np.float32,
                      "is_terminal": np.bool_,
                      "is_first": np.bool_,
                      "is_last": np.bool_,
                      "discount": np.float32,
                  }
              ),
              "episode_id": np.int64,
              "timestamp": np.int64,
          }
      )
  
file_path: datasets/tensorflow_datasets/rl_unplugged/rlu_control_suite/rlu_control_suite.py
imports: from __future__ import annotations
  import dataclasses
  from typing import Any, Dict, Optional
  import numpy
  from tensorflow_datasets.core.utils.lazy_imports_utils import tensorflow
  import tensorflow_datasets.public_api
  from tensorflow_datasets.rl_unplugged import rlu_common
label_desc: Use image node_id 1 for calling the RluControlSuite obj's underlying member method code with example usage: obj.get_features_dict() and returns: tfds