Minari Utils

agilerl.utils.minari_utils.load_minari_dataset(dataset_id: str, accelerator: Accelerator | None = None, remote: bool = False) → MinariDataset

Load a Minari dataset either from local storage or remote repository.

Parameters:

  • dataset_id (str) – The ID of the Minari dataset to load

  • accelerator (Accelerator | None) – Optional accelerator for distributed training

  • remote – Whether to load from remote repository. Defaults to False.

Returns:

The loaded Minari dataset

Raises:

  • KeyError – If remote=True and dataset_id is not a valid remote dataset

  • FileNotFoundError – If remote=False and dataset not found locally

agilerl.utils.minari_utils.minari_to_agile_buffer(dataset_id: str, memory: ReplayBuffer, accelerator: Accelerator | None = None, remote: bool = False) → ReplayBuffer

Convert a Minari dataset to an agile buffer.

Parameters:

  • dataset_id – The ID of the Minari dataset to load

  • memory – The memory to save the dataset to

  • accelerator – Optional accelerator for distributed training

  • remote – Whether to load from remote repository. Defaults to False.

Returns:

The loaded Minari dataset

agilerl.utils.minari_utils.minari_to_agile_dataset(dataset_id: str, remote: bool = False) → File

Convert a Minari dataset to an agile dataset.

Parameters:

  • dataset_id – The ID of the Minari dataset to load

  • remote – Whether to load from remote repository. Defaults to False.

Returns:

The loaded Minari dataset