fix InvertedPendulum

openai · May 1, 2016 · 7b91967 · 7b91967
1 parent 795640d
commit 7b91967
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gym/envs/mujoco/inverted_pendulum.py b/gym/envs/mujoco/inverted_pendulum.py
@@ -16,8 +16,8 @@ def _step(self, a):
         return ob, reward, done, {}
 
     def reset_model(self):
-        qpos = self.init_qpos + np.random.uniform(size=(self.model.nq,1), low=-0.01, high=0.01)
-        qvel = self.init_qvel + np.random.uniform(size=(self.model.nv,1), low=-0.01, high=0.01)
+        qpos = self.init_qpos + np.random.uniform(size=self.model.nq, low=-0.01, high=0.01)
+        qvel = self.init_qvel + np.random.uniform(size=self.model.nv, low=-0.01, high=0.01)
         self.set_state(qpos, qvel)
         return self._get_obs()