fix discrete versions of kuka grasping environment

should fix issue 1386
2017-10-31 16:14:20 -07:00
parent 33d32d7ab4
commit 3709e68645
3 changed files with 3 additions and 3 deletions
--- a/examples/pybullet/gym/pybullet_envs/baselines/train_kuka_grasping.py
+++ b/examples/pybullet/gym/pybullet_envs/baselines/train_kuka_grasping.py
@@ -25,7 +25,7 @@ def callback(lcl, glb):

 def main():
  	
-    env = KukaGymEnv(renders=False)
+    env = KukaGymEnv(renders=False, isDiscrete=True)
    model = deepq.models.mlp([64])
    act = deepq.learn(
        env,