Regarding data download

[jayLEE0301]

Hello @atharvamete, I appreciate your effort to release the codebase.

when I tired

python scripts/generate_metaworld_dataset.py

I found the error below

Available options in 'task':
        libero_90
        libero_base
        libero_long
        libero_long_fewshot
        metaworld_base
        metaworld_ml45_prise
        metaworld_ml45_prise_fewshot

so I changed metaworld_ml45 to metaworld_base to solve this error, as shown below (in collect_data.yaml)

defaults:
  - task: metaworld_base # originally metaworld_ml45
  - algo: data_collect
  - _self_

however, even after changing this, I got another issue:

omegaconf.errors.InterpolationKeyError: Interpolation key 'task.n_tasks' not found

and I found that there's no task.n_tasks in metaworld_base.yaml

could you please give me some directions that I can solve this issue and download the dataset using python scripts/generate_metaworld_dataset.py?

thank you, Seungjae.

[atharvamete]

Hey @jayLEE0301 The base yaml files are used by other yaml files where the n_tasks is defined respectively.

I've just added the metaworld_ml45.yaml. This basically uses original test tasks from metaworld. Since PRISE uses a different set of test tasks we also give metaworld_ml45_prise.yaml. Note that all our results are based on PRISE version.

I've updated the repo, and tested the data gen script on my end. It should be working fine now. You can cherry pick cac6bb76090aa20e3bfb643f6ba74033f78991f1

Thanks for pointing this out!

[jayLEE0301]

Thank you for your reply! let me close this issue.