search

ARISE-Initiative / robosuite

robosuite: A Modular Simulation Framework and Benchmark for Robot Learning
https://robosuite.ai
Other
1.23k stars 394 forks source link

Add custom env to robosuite.make #468

Open bhe1004 opened 5 months ago

bhe1004 commented 5 months ago

How can I create a custom env through robosuite.make? I have created a single_arm_peg_in_hole env like the format of the env python files in robosuite's environments/minipulatation, but I don't know how to apply it to robosuite.make().

zhuyifengzju commented 4 months ago

Hi, if you want robosuite.make() to find it, you would need to import your custom env before (ALL_ENVIRONMENTS)[https://github.com/ARISE-Initiative/robosuite/blob/eafb81f54ffc104f905ee48a16bb15f059176ad3/robosuite/environments/__init__.py#L3] is set. Look into this variable to see if your custom environment has been included.