From 3812c034c9cf084b8e45d41ce6938613ec22fb2a Mon Sep 17 00:00:00 2001
From: kngwyu <yuji.kanagawa@oist.jp>
Date: Thu, 27 May 2021 01:06:06 +0900
Subject: [PATCH] Fix NoRewardSquareRoom.reward

---
 mujoco_maze/maze_task.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mujoco_maze/maze_task.py b/mujoco_maze/maze_task.py
index 4ee801b..9e102a9 100644
--- a/mujoco_maze/maze_task.py
+++ b/mujoco_maze/maze_task.py
@@ -168,7 +168,7 @@ class NoRewardSquareRoom(GoalRewardSimpleRoom):
     def __init__(self, scale: float) -> None:
         super().__init__(scale)
 
-    def reward(self) -> float:
+    def reward(self, _obs: np.ndarray) -> float:
         return 0.0