diff --git a/examples/pybullet/gym/pybullet_envs/__init__.py b/examples/pybullet/gym/pybullet_envs/__init__.py index def497392..a605dafc4 100644 --- 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',