Skip to content

Commit 2d08055

Browse files
sythamdementrock
authored andcommitted
Fixes issue #111 (#112)
1 parent 2387f4c commit 2d08055

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

rllab/envs/mujoco/mujoco_env.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -197,15 +197,15 @@ def get_viewer(self):
197197
self.viewer.set_model(self.model)
198198
return self.viewer
199199

200-
def render(self, close=False, mode='human', config=None):
200+
def render(self, close=False, mode='human'):
201201
if mode == 'human':
202-
viewer = self.get_viewer(config=config)
202+
viewer = self.get_viewer()
203203
viewer.loop_once()
204204
elif mode == 'rgb_array':
205-
viewer = self.get_viewer(config=config)
205+
viewer = self.get_viewer()
206206
viewer.loop_once()
207207
# self.get_viewer(config=config).render()
208-
data, width, height = self.get_viewer(config=config).get_image()
208+
data, width, height = self.get_viewer().get_image()
209209
return np.fromstring(data, dtype='uint8').reshape(height, width, 3)[::-1,:,:]
210210
if close:
211211
self.stop_viewer()

0 commit comments

Comments
 (0)