diff --git a/examples/pybullet/gym/pybullet_envs/bullet/kuka_diverse_object_gym_env.py b/examples/pybullet/gym/pybullet_envs/bullet/kuka_diverse_object_gym_env.py index 8ed46fbe0..c2c4d2acc 100644 --- a/examples/pybullet/gym/pybullet_envs/bullet/kuka_diverse_object_gym_env.py +++ b/examples/pybullet/gym/pybullet_envs/bullet/kuka_diverse_object_gym_env.py @@ -316,7 +316,7 @@ class KukaDiverseObjectEnv(KukaGymEnv): if test: urdf_pattern = os.path.join(self._urdfRoot, 'random_urdfs/*0/*.urdf') else: - urdf_pattern = os.path.join(self._urdfRoot, 'random_urdfs/*[^0]/*.urdf') + urdf_pattern = os.path.join(self._urdfRoot, 'random_urdfs/*[1-9]/*.urdf') found_object_directories = glob.glob(urdf_pattern) total_num_objects = len(found_object_directories) selected_objects = np.random.choice(np.arange(total_num_objects),