PyBullet: register KukaDiverseObjectGrasping-v0 to Gym.

2018-01-24 18:27:03 -08:00
parent 4ea406c73c
commit 5d45fd5ca5
1 changed files with 7 additions and 0 deletions
--- a/examples/pybullet/gym/pybullet_envs/init.py
+++ b/examples/pybullet/gym/pybullet_envs/init.py
@@ -59,6 +59,13 @@ register(
 	reward_threshold=5.0,
 )

+register(
+	id='KukaDiverseObjectGrasping-v0',
+	entry_point='pybullet_envs.bullet:KukaDiverseObjectEnv',
+	timestep_limit=1000,
+	reward_threshold=5.0,
+)
+
 register(
 	id='InvertedPendulumBulletEnv-v0',
 	entry_point='pybullet_envs.gym_pendulum_envs:InvertedPendulumBulletEnv',