Skip to content

Commit cc15f6c

Browse files
committed
fixed reward callback
1 parent 926a161 commit cc15f6c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

multiagent/environment.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,7 @@ def _get_obs(self, agent):
129129
def _get_reward(self, agent):
130130
if self.reward_callback is None:
131131
return 0.0
132-
self.reward_callback(agent, self.world)
132+
return self.reward_callback(agent, self.world)
133133

134134
# set env action for a particular agent
135135
def _set_action(self, action, agent, action_space, time=None):

0 commit comments

Comments
 (0)