Bart/bullet3
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #2159 from bulletphysics/erwincoumans-glob

Browse Source 
Update kuka_diverse_object_gym_env.py
This commit is contained in:
erwincoumans
2019-03-14 18:28:14 -07:00
committed by GitHub
parent ced17be7f3 9662ae4ba0
commit 17d2b318bf
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pybullet/gym/pybullet_envs/bullet/kuka_diverse_object_gym_env.py
View File

@@ -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),