Gets the keys present in the D4RL dataset.
(h5file)
| 246 | |
| 247 | |
| 248 | def _get_dataset_keys(h5file): |
| 249 | """Gets the keys present in the D4RL dataset.""" |
| 250 | keys = [] |
| 251 | |
| 252 | def visitor(name, item): |
| 253 | if isinstance(item, h5py.Dataset): |
| 254 | keys.append(name) |
| 255 | |
| 256 | h5file.visititems(visitor) |
| 257 | return keys |
| 258 | |
| 259 | |
| 260 | def read_d4rl_dataset(file_path: str): |