2017-03-10 19:22:38 +00:00
|
|
|
from gym.envs.registration import registry, register, make, spec
|
|
|
|
|
|
|
|
# ------------bullet-------------
|
|
|
|
|
2018-01-15 20:48:32 +00:00
|
|
|
register(
|
|
|
|
id='CartPoleBulletEnv-v0',
|
|
|
|
entry_point='pybullet_envs.bullet:CartPoleBulletEnv',
|
|
|
|
timestep_limit=1000,
|
|
|
|
reward_threshold=950.0,
|
|
|
|
)
|
|
|
|
|
2017-03-10 19:22:38 +00:00
|
|
|
register(
|
|
|
|
id='MinitaurBulletEnv-v0',
|
2017-08-22 07:42:02 +00:00
|
|
|
entry_point='pybullet_envs.bullet:MinitaurBulletEnv',
|
2017-03-10 19:22:38 +00:00
|
|
|
timestep_limit=1000,
|
|
|
|
reward_threshold=5.0,
|
|
|
|
)
|
2017-06-09 02:45:48 +00:00
|
|
|
|
2017-10-31 23:05:52 +00:00
|
|
|
register(
|
|
|
|
id='MinitaurBulletDuckEnv-v0',
|
|
|
|
entry_point='pybullet_envs.bullet:MinitaurBulletDuckEnv',
|
|
|
|
timestep_limit=1000,
|
|
|
|
reward_threshold=5.0,
|
|
|
|
)
|
|
|
|
|
|
|
|
|
2017-06-09 02:45:48 +00:00
|
|
|
register(
|
|
|
|
id='RacecarBulletEnv-v0',
|
2017-08-25 05:01:45 +00:00
|
|
|
entry_point='pybullet_envs.bullet:RacecarGymEnv',
|
2017-06-09 02:45:48 +00:00
|
|
|
timestep_limit=1000,
|
|
|
|
reward_threshold=5.0,
|
|
|
|
)
|
2017-06-13 23:04:50 +00:00
|
|
|
|
|
|
|
register(
|
2018-01-15 20:48:32 +00:00
|
|
|
id='RacecarZedBulletEnv-v0',
|
|
|
|
entry_point='pybullet_envs.bullet:RacecarZEDGymEnv',
|
|
|
|
timestep_limit=1000,
|
|
|
|
reward_threshold=5.0,
|
2017-06-13 23:04:50 +00:00
|
|
|
)
|
2017-06-14 07:54:41 +00:00
|
|
|
|
2017-08-16 22:06:01 +00:00
|
|
|
register(
|
2018-01-15 20:48:32 +00:00
|
|
|
id='SimpleHumanoidBulletEnv-v0',
|
|
|
|
entry_point='pybullet_envs.bullet:SimpleHumanoidGymEnv',
|
|
|
|
timestep_limit=1000,
|
|
|
|
reward_threshold=5.0,
|
2017-08-16 22:06:01 +00:00
|
|
|
)
|
2017-06-14 07:54:41 +00:00
|
|
|
|
|
|
|
register(
|
2018-01-15 20:48:32 +00:00
|
|
|
id='KukaBulletEnv-v0',
|
|
|
|
entry_point='pybullet_envs.bullet:KukaGymEnv',
|
|
|
|
timestep_limit=1000,
|
|
|
|
reward_threshold=5.0,
|
2017-06-21 16:33:46 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
register(
|
2018-01-15 20:48:32 +00:00
|
|
|
id='KukaCamBulletEnv-v0',
|
|
|
|
entry_point='pybullet_envs.bullet:KukaCamGymEnv',
|
|
|
|
timestep_limit=1000,
|
|
|
|
reward_threshold=5.0,
|
2017-08-15 13:31:24 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
register(
|
|
|
|
id='InvertedPendulumBulletEnv-v0',
|
2017-08-22 07:42:02 +00:00
|
|
|
entry_point='pybullet_envs.gym_pendulum_envs:InvertedPendulumBulletEnv',
|
2017-08-15 13:31:24 +00:00
|
|
|
max_episode_steps=1000,
|
|
|
|
reward_threshold=950.0,
|
|
|
|
)
|
|
|
|
|
|
|
|
register(
|
|
|
|
id='InvertedDoublePendulumBulletEnv-v0',
|
2017-08-22 07:42:02 +00:00
|
|
|
entry_point='pybullet_envs.gym_pendulum_envs:InvertedDoublePendulumBulletEnv',
|
2017-08-15 13:31:24 +00:00
|
|
|
max_episode_steps=1000,
|
|
|
|
reward_threshold=9100.0,
|
|
|
|
)
|
|
|
|
|
|
|
|
register(
|
|
|
|
id='InvertedPendulumSwingupBulletEnv-v0',
|
2017-08-22 07:42:02 +00:00
|
|
|
entry_point='pybullet_envs.gym_pendulum_envs:InvertedPendulumSwingupBulletEnv',
|
2017-08-15 13:31:24 +00:00
|
|
|
max_episode_steps=1000,
|
|
|
|
reward_threshold=800.0,
|
|
|
|
)
|
|
|
|
|
2018-01-15 20:48:32 +00:00
|
|
|
register(
|
|
|
|
id='ReacherBulletEnv-v0',
|
|
|
|
entry_point='pybullet_envs.gym_manipulator_envs:ReacherBulletEnv',
|
|
|
|
max_episode_steps=150,
|
|
|
|
reward_threshold=18.0,
|
|
|
|
)
|
|
|
|
|
|
|
|
register(
|
|
|
|
id='PusherBulletEnv-v0',
|
|
|
|
entry_point='pybullet_envs.gym_manipulator_envs:PusherBulletEnv',
|
|
|
|
max_episode_steps=150,
|
|
|
|
reward_threshold=18.0,
|
|
|
|
)
|
|
|
|
|
|
|
|
register(
|
|
|
|
id='ThrowerBulletEnv-v0',
|
|
|
|
entry_point='pybullet_envs.gym_manipulator_envs:ThrowerBulletEnv',
|
|
|
|
max_episode_steps=100,
|
|
|
|
reward_threshold=18.0,
|
|
|
|
)
|
|
|
|
|
|
|
|
register(
|
|
|
|
id='StrikerBulletEnv-v0',
|
|
|
|
entry_point='pybullet_envs.gym_manipulator_envs:StrikerBulletEnv',
|
|
|
|
max_episode_steps=100,
|
|
|
|
reward_threshold=18.0,
|
|
|
|
)
|
2017-08-15 13:31:24 +00:00
|
|
|
|
|
|
|
register(
|
|
|
|
id='Walker2DBulletEnv-v0',
|
2017-08-22 07:42:02 +00:00
|
|
|
entry_point='pybullet_envs.gym_locomotion_envs:Walker2DBulletEnv',
|
2017-08-15 13:31:24 +00:00
|
|
|
max_episode_steps=1000,
|
|
|
|
reward_threshold=2500.0
|
|
|
|
)
|
|
|
|
register(
|
|
|
|
id='HalfCheetahBulletEnv-v0',
|
2017-08-22 07:42:02 +00:00
|
|
|
entry_point='pybullet_envs.gym_locomotion_envs:HalfCheetahBulletEnv',
|
2017-08-15 13:31:24 +00:00
|
|
|
max_episode_steps=1000,
|
|
|
|
reward_threshold=3000.0
|
|
|
|
)
|
|
|
|
|
|
|
|
register(
|
|
|
|
id='AntBulletEnv-v0',
|
2017-08-22 07:42:02 +00:00
|
|
|
entry_point='pybullet_envs.gym_locomotion_envs:AntBulletEnv',
|
2017-08-15 13:31:24 +00:00
|
|
|
max_episode_steps=1000,
|
|
|
|
reward_threshold=2500.0
|
|
|
|
)
|
|
|
|
|
2018-01-15 20:48:32 +00:00
|
|
|
register(
|
|
|
|
id='HopperBulletEnv-v0',
|
|
|
|
entry_point='pybullet_envs.gym_locomotion_envs:HopperBulletEnv',
|
|
|
|
max_episode_steps=1000,
|
|
|
|
reward_threshold=2500.0
|
|
|
|
)
|
|
|
|
|
2017-08-15 13:31:24 +00:00
|
|
|
register(
|
|
|
|
id='HumanoidBulletEnv-v0',
|
2017-08-22 07:42:02 +00:00
|
|
|
entry_point='pybullet_envs.gym_locomotion_envs:HumanoidBulletEnv',
|
2017-08-15 13:31:24 +00:00
|
|
|
max_episode_steps=1000
|
|
|
|
)
|
2017-08-15 19:33:55 +00:00
|
|
|
|
|
|
|
register(
|
2018-01-15 20:48:32 +00:00
|
|
|
id='HumanoidFlagrunBulletEnv-v0',
|
|
|
|
entry_point='pybullet_envs.gym_locomotion_envs:HumanoidFlagrunBulletEnv',
|
2017-08-15 19:33:55 +00:00
|
|
|
max_episode_steps=1000,
|
2018-01-15 20:48:32 +00:00
|
|
|
reward_threshold=2000.0
|
2017-08-22 07:42:02 +00:00
|
|
|
)
|
2017-08-31 16:59:40 +00:00
|
|
|
|
2018-01-15 20:48:32 +00:00
|
|
|
register(
|
|
|
|
id='HumanoidFlagrunHarderBulletEnv-v0',
|
|
|
|
entry_point='pybullet_envs.gym_locomotion_envs:HumanoidFlagrunHarderBulletEnv',
|
|
|
|
max_episode_steps=1000
|
|
|
|
)
|
|
|
|
|
|
|
|
#register(
|
|
|
|
# id='AtlasBulletEnv-v0',
|
|
|
|
# entry_point='pybullet_envs.gym_locomotion_envs:AtlasBulletEnv',
|
|
|
|
# max_episode_steps=1000
|
|
|
|
# )
|
|
|
|
|
|
|
|
def getList():
|
|
|
|
btenvs = ['- ' + spec.id for spec in gym.envs.registry.all() if spec.id.find('Bullet')>=0]
|
|
|
|
return btenvs
|