diff --git a/examples/pybullet/gym/pybullet_envs/scene_stadium.py b/examples/pybullet/gym/pybullet_envs/scene_stadium.py index 8aef2552c..c105983db 100644 --- a/examples/pybullet/gym/pybullet_envs/scene_stadium.py +++ b/examples/pybullet/gym/pybullet_envs/scene_stadium.py @@ -17,7 +17,7 @@ class StadiumScene(Scene): def episode_restart(self, bullet_client): self._p = bullet_client - Scene.episode_restart(self) # contains cpp_world.clean_everything() + Scene.episode_restart(self, bullet_client) # contains cpp_world.clean_everything() if (self.stadiumLoaded==0): self.stadiumLoaded=1