From 8928975f8439a6920791e099b502f2dd49b637ff Mon Sep 17 00:00:00 2001 From: Diego Ferigo Date: Mon, 16 Sep 2019 17:30:23 +0200 Subject: [PATCH] Updated environments kwargs --- gym_ignition/__init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/gym_ignition/__init__.py b/gym_ignition/__init__.py index f1d90a946..a822039a8 100644 --- a/gym_ignition/__init__.py +++ b/gym_ignition/__init__.py @@ -44,8 +44,8 @@ id='CartPoleGymppy-Discrete-v0', entry_point='gym_ignition.base.gazebo_env:GazeboEnv', max_episode_steps=5000, - kwargs={'task': cartpole_discrete.CartPoleDiscrete, - 'robot': sim.cartpole.CartPoleRobot, + kwargs={'task_cls': cartpole_discrete.CartPoleDiscrete, + 'robot_cls': sim.cartpole.CartPoleRobot, 'sdf': "CartPole/CartPole.sdf", 'world': "DefaultEmptyWorld.world", 'rtf': max_float, @@ -57,8 +57,8 @@ id='CartPoleGymppy-Continuous-v0', entry_point='gym_ignition.base.gazebo_env:GazeboEnv', max_episode_steps=5000, - kwargs={'task': cartpole_continuous.CartPoleContinuous, - 'robot': sim.cartpole.CartPoleRobot, + kwargs={'task_cls': cartpole_continuous.CartPoleContinuous, + 'robot_cls': sim.cartpole.CartPoleRobot, 'sdf': "CartPole/CartPole.sdf", 'world': "DefaultEmptyWorld.world", 'rtf': max_float,