From de1a1159c473a81374c79cd597f34473de3b9f59 Mon Sep 17 00:00:00 2001 From: Ma Haozhe Date: Sun, 17 Sep 2023 15:08:50 +0800 Subject: [PATCH] fixed a typo in core.py --- panda_gym/envs/core.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/panda_gym/envs/core.py b/panda_gym/envs/core.py index 2004b8cf..fdaffe0e 100644 --- a/panda_gym/envs/core.py +++ b/panda_gym/envs/core.py @@ -233,12 +233,12 @@ def __init__( observation, _ = self.reset() # required for init; seed can be changed later observation_shape = observation["observation"].shape achieved_goal_shape = observation["achieved_goal"].shape - desired_goal_shape = observation["achieved_goal"].shape + desired_goal_shape = observation["desired_goal"].shape self.observation_space = spaces.Dict( dict( observation=spaces.Box(-10.0, 10.0, shape=observation_shape, dtype=np.float32), - desired_goal=spaces.Box(-10.0, 10.0, shape=achieved_goal_shape, dtype=np.float32), - achieved_goal=spaces.Box(-10.0, 10.0, shape=desired_goal_shape, dtype=np.float32), + desired_goal=spaces.Box(-10.0, 10.0, shape=desired_goal_shape, dtype=np.float32), + achieved_goal=spaces.Box(-10.0, 10.0, shape=achieved_goal_shape, dtype=np.float32), ) ) self.action_space = self.robot.action_space