dodox/mujoco_maze
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Change the start position of SquareRoom

Browse Source
This commit is contained in:
kngwyu 2021-05-28 11:44:11 +09:00
parent 3812c034c9
commit ddb8a5f3c0
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mujoco_maze/maze_task.py
View File

@ -148,7 +148,7 @@ class DistRewardSimpleRoom(GoalRewardSimpleRoom, DistRewardMixIn):
class GoalRewardSquareRoom(GoalRewardUMaze): class GoalRewardSquareRoom(GoalRewardUMaze):
MAZE_SIZE_SCALING: Scaling = Scaling(4.0, 4.0, 2.0) MAZE_SIZE_SCALING: Scaling = Scaling(4.0, 4.0, 2.0)
def __init__(self, scale: float, goal: Tuple[float, float] = (2.0, 0.0)) -> None: def __init__(self, scale: float, goal: Tuple[float, float] = (1.0, 0.0)) -> None:
super().__init__(scale) super().__init__(scale)
self.goals = [MazeGoal(np.array(goal) * scale)] self.goals = [MazeGoal(np.array(goal) * scale)]
@ -158,7 +158,7 @@ class GoalRewardSquareRoom(GoalRewardUMaze):
return [ return [
[B, B, B, B, B], [B, B, B, B, B],
[B, E, E, E, B], [B, E, E, E, B],
[B, R, E, E, B], [B, E, R, E, B],
[B, E, E, E, B], [B, E, E, E, B],
[B, B, B, B, B], [B, B, B, B, B],
] ]