d3rlpy.datasets.get_minari

d3rlpy.datasets.get_minari(env_name, transition_picker=None, trajectory_slicer=None, render_mode=None, tuple_observation=False)

Returns minari dataset and envrironment.

The dataset is provided through minari.

Parameters

• env_name (str) – environment id of minari dataset.

• transition_picker (Optional[d3rlpy.dataset.transition_pickers.TransitionPickerProtocol]) – TransitionPickerProtocol object.

• trajectory_slicer (Optional[d3rlpy.dataset.trajectory_slicers.TrajectorySlicerProtocol]) – TrajectorySlicerProtocol object.

• render_mode (Optional[str]) – Mode of rendering (human, rgb_array).

• tuple_observation (bool) – Flag to include goals as tuple element.

Returns

tuple of d3rlpy.dataset.ReplayBuffer and gym environment.

Return type

Tuple[d3rlpy.dataset.replay_buffer.ReplayBuffer, gymnasium.core.Env[Any, Any]]