diff --git a/gym/data/download_d4rl_datasets.py b/gym/data/download_d4rl_datasets.py index 5751ceeb..8dde33ae 100644 --- a/gym/data/download_d4rl_datasets.py +++ b/gym/data/download_d4rl_datasets.py @@ -10,7 +10,7 @@ datasets = [] for env_name in ['halfcheetah', 'hopper', 'walker2d']: - for dataset_type in ['medium', 'medium-replay', 'expert']: + for dataset_type in ['medium-expert', 'medium', 'medium-replay', 'expert']: name = f'{env_name}-{dataset_type}-v2' env = gym.make(name) dataset = env.get_dataset()