-
Notifications
You must be signed in to change notification settings - Fork 9
Description
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!