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

NoRewardMultiPush

Browse Source
This commit is contained in:
Yuji Kanagawa 2021-09-24 22:07:42 +09:00
parent 2e0c9da3d1
commit 858d5ebcc0
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mujoco_maze/maze_task.py
View File

@ -225,6 +225,11 @@ class DistRewardMultiPush(GoalRewardMultiPush, DistRewardMixIn):
pass pass
class NoRewardMultiPush(GoalRewardMultiPush):
def reward(self, _obs: np.ndarray) -> float:
return 0.0
class GoalRewardFall(GoalRewardUMaze): class GoalRewardFall(GoalRewardUMaze):
OBSERVE_BLOCKS: bool = True OBSERVE_BLOCKS: bool = True
@ -587,7 +592,7 @@ class TaskRegistry:
"SquareRoom": [DistRewardSquareRoom, GoalRewardSquareRoom, NoRewardSquareRoom], "SquareRoom": [DistRewardSquareRoom, GoalRewardSquareRoom, NoRewardSquareRoom],
"UMaze": [DistRewardUMaze, GoalRewardUMaze], "UMaze": [DistRewardUMaze, GoalRewardUMaze],
"Push": [DistRewardPush, GoalRewardPush], "Push": [DistRewardPush, GoalRewardPush],
"MultiPush": [DistRewardMultiPush, GoalRewardMultiPush], "MultiPush": [DistRewardMultiPush, GoalRewardMultiPush, NoRewardMultiPush],
"Fall": [DistRewardFall, GoalRewardFall], "Fall": [DistRewardFall, GoalRewardFall],
"2Rooms": [DistReward2Rooms, GoalReward2Rooms, SubGoal2Rooms], "2Rooms": [DistReward2Rooms, GoalReward2Rooms, SubGoal2Rooms],
"4Rooms": [DistReward4Rooms, GoalReward4Rooms, SubGoal4Rooms], "4Rooms": [DistReward4Rooms, GoalReward4Rooms, SubGoal4Rooms],