Updated environments kwargs

robotology · Sep 19, 2019 · 8928975 · 8928975
1 parent 9883023
commit 8928975
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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,