ModuleNotFoundError: No module named 'locomotion'

[Stone]

When I want to use gym.
env = gym.make('motion_imitation: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'

[Stone]

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,
)