erwincoumans / motion_imitation

Code accompanying the paper "Learning Agile Robotic Locomotion Skills by Imitating Animals"
Apache License 2.0
1.18k stars 292 forks source link

ModuleNotFoundError: No module named 'locomotion' #45

Closed HaojieSHI98 closed 4 years ago

HaojieSHI98 commented 4 years ago

When I want to use gym. _env = gym.make('motionimitation:A1GymEnv-v1', render=True) Error Occurred. I dont' know what's wrong with this, and wonder if someone could help fix it.

Traceback (most recent call last): File "", line 1, in File "/home/robot/anaconda2/envs/stone_legged/lib/python3.6/site-packages/gym/envs/registration.py", line 142, in make return registry.make(id, kwargs) File "/home/robot/anaconda2/envs/stone_legged/lib/python3.6/site-packages/gym/envs/registration.py", line 87, in make env = spec.make(kwargs) File "/home/robot/anaconda2/envs/stone_legged/lib/python3.6/site-packages/gym/envs/registration.py", line 58, in make cls = load(self.entry_point) File "/home/robot/anaconda2/envs/stone_legged/lib/python3.6/site-packages/gym/envs/registration.py", line 17, in load mod = importlib.import_module(mod_name) File "/home/robot/anaconda2/envs/stone_legged/lib/python3.6/importlib/init.py", line 126, in import_module return _bootstrap._gcd_import(name[level:], package, level) File "", line 994, in _gcd_import File "", line 971, in _find_and_load File "", line 941, in _find_and_load_unlocked File "", line 219, in _call_with_frames_removed File "", line 994, in _gcd_import File "", line 971, in _find_and_load File "", line 941, in _find_and_load_unlocked File "", line 219, in _call_with_frames_removed File "", line 994, in _gcd_import File "", line 971, in _find_and_load File "", line 953, in _find_and_load_unlocked ModuleNotFoundError: No module named 'locomotion'

HaojieSHI98 commented 4 years ago

I have solved it, just modify the /motion_imitation/init.py as register( env_id='A1GymEnv-v1', entry_point='motion_imitation.envs.gym_envs:A1GymEnv', max_episode_steps=2000, reward_threshold=2000.0, )