Skip to content

Missing no_reward task_generator_id in CausalWorld (used in dataset generation script) #4

@Slian22

Description

@Slian22

Hi, thanks for releasing this great environment!

While running the official data_generation_causal_world.py script from [your repository / paper], I ran into the following issue:

CUDA_VISIBLE_DEVICES=1 python data_generation/data_generation_causal_world.py
--output_folder data/causal_world
--num_sequences 200

produces:

Exception: No valid task_generator_id
File ".../task.py", line 44, in generate_task
raise Exception("No valid task_generator_id")

In the script, create_environment() calls:

task = generate_task(task_generator_id='no_reward')

but according to generate_task(...), valid task_generator_id values are only:
picking, pushing, reaching, pick_and_place, stacking2, stacked_blocks, towers, general, creative_stacked_blocks.

It seems that 'no_reward' is not a valid task anymore (maybe removed in a recent version?).
My questions are:
1. What is the recommended replacement for no_reward?
Should I simply switch to pushing (and manually set reward to zero), or is there a dedicated “no_reward” version of tasks somewhere else?
2. For reproducing the dataset generation from the paper, which task_generator_id should I use to get consistent results?

Thanks a lot for your help!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions