diff options
| author | erwincoumans <erwin.coumans@gmail.com> | 2019-03-14 09:33:48 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2019-03-14 09:33:48 -0700 |
| commit | 9662ae4ba0f27fa42757af85a7fb07ff0650b529 (patch) | |
| tree | aa6aed0eb0b118513fccf55158f0fa3b81c923c3 | |
| parent | 18918389d3076637b0f19793e607b28fba049d53 (diff) | |
| download | bullet3-erwincoumans-glob.tar.gz | |
Update kuka_diverse_object_gym_env.pyerwincoumans-glob
Fix issue in glob to separate training/test data pattern. See https://github.com/bulletphysics/bullet3/issues/2158
Thank to @TomoyaFukui
| -rw-r--r-- | examples/pybullet/gym/pybullet_envs/bullet/kuka_diverse_object_gym_env.py | 2 |
1 files changed, 1 insertions, 1 deletions
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), |