Issues with generating ant maze dataset

[dh2shin]

Hi, I'm trying to generate the ant maze dataset using the generation script but getting a "No module named 'locomotion.ant'" error when loading the policy load_policy('ant_hierarch_pol.pkl'). I installed the locomotion module as well as the rlkit fork, so any idea on how to fix this error?

[louieworth]

I am struggling with the same issue.

[CUN-bjy]

I have the same issue now.

[gaisibo]

I find that the locomotion here is not the locomotion module installed by pip but the locomotion dir of d4rl. Make a copy of d4rl/locomotion to the root dir and change the import in it (change import d4rl.locomotion.* to import locomotion.*). Remember to remove the environment register in the locomotion/init.py to fix the re-register question.

[tlatjddnd101]

@gaisibo Thanks! It works.