DESC = '''
Helper script to load an example dataset in Roboset.\n
USAGE:\n
    $ python example_load.py \n
Sample real robot data: baking_sample_data.h5
Sample simulation data: sim_sample.h5
'''

from robohive.logger.grouped_datasets import Trace
filename = 'sim_sample.h5'
h5_paths = Trace.load(filename)
i = 0
trial_i = h5_paths[f'Trial{i}']
obs = trial_i['observations'][...]
act = trial_i['actions'][...]
rwd = trial_i['rewards'][...]
env_infos = trial_i['env_infos']
