Fix gym env following action changes.

stepjam · Apr 2, 2020 · 8ff9507 · 8ff9507
1 parent 0897b7b
commit 8ff9507
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rlbench/gym/rlbench_env.py b/rlbench/gym/rlbench_env.py
@@ -38,7 +38,7 @@ def __init__(self, task_class, observation_mode='state',
         _, obs = self.task.reset()
 
         self.action_space = spaces.Box(
-            low=-1.0, high=1.0, shape=(action_mode.action_size,))
+            low=-1.0, high=1.0, shape=(self.env.action_size,))
 
         if observation_mode == 'state':
             self.observation_space = spaces.Box(