From 9662ae4ba0f27fa42757af85a7fb07ff0650b529 Mon Sep 17 00:00:00 2001 From: erwincoumans Date: Thu, 14 Mar 2019 09:33:48 -0700 Subject: [PATCH] Update kuka_diverse_object_gym_env.py Fix issue in glob to separate training/test data pattern. See https://github.com/bulletphysics/bullet3/issues/2158 Thank to @TomoyaFukui --- .../gym/pybullet_envs/bullet/kuka_diverse_object_gym_env.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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),