Skip to content

Commit 8ff9507

Browse files
committed
Fix gym env following action changes.
1 parent 0897b7b commit 8ff9507

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

rlbench/gym/rlbench_env.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def __init__(self, task_class, observation_mode='state',
3838
_, obs = self.task.reset()
3939

4040
self.action_space = spaces.Box(
41-
low=-1.0, high=1.0, shape=(action_mode.action_size,))
41+
low=-1.0, high=1.0, shape=(self.env.action_size,))
4242

4343
if observation_mode == 'state':
4444
self.observation_space = spaces.Box(

0 commit comments

Comments
 (0)