mimicgen package#

mimicgen.register_all_links()#

Record all dataset links in this function.

mimicgen.register_dataset_link(dataset_type, task, link, horizon)#

Helper function to register dataset link in global dictionary. Also takes a @horizon parameter - this corresponds to the evaluation rollout horizon that should be used during training.

Parameters:

• dataset_type (str) – identifies the type of dataset (e.g. source human data, core experiment data, object transfer data)

• task (str) – name of task for this dataset

• link (str) – download link for the dataset

• horizon (int) – evaluation rollout horizon that should be used with this dataset