Dataset Adapters

Dataset adapters for external offline RL benchmarks.

load_d4rl_dataset(env_name, *, discount=0.99, include_timeouts=True, return_type='transition')

Load a D4RL dataset and map it to a TransitionDataset.

load_rl_unplugged_dataset(dataset_name, *, split='train', discount=0.99, return_type='transition', max_episodes=None)

Load an RL Unplugged dataset and map it to a TransitionDataset.