d3rlpy.datasets.get_pybullet

d3rlpy.datasets.get_pybullet(env_name, as_tensor=False, device=None)

Returns pybullet dataset and envrironment.

The dataset is provided through d4rl-pybullet. See more details including available dataset from its GitHub page.

from d3rlpy.datasets import get_pybullet

dataset, env = get_pybullet('hopper-bullet-mixed-v0')

References

• https://github.com/takuseno/d4rl-pybullet

Parameters:

• env_name (str) – environment id of d4rl-pybullet dataset.
• as_tensor (bool) – flag to hold observations as torch.Tensor.
• device (d3rlpy.gpu.Device or int) – gpu device or device id for tensor.

Returns:

tuple of d3rlpy.dataset.MDPDataset and gym environment.

Return type:

tuple